Package Dependency

[0xga]

@xh9998 Thanks for your contribution.
Inferencing doesn't work on H100 machine with current versions of packages.

Loading pipeline components...: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 6/6 [00:00<00:00, 6.95it/s]/home/ubuntu/miniconda3/envs/DiffVSR/lib/python3.9/site-packages/torch/cuda/init.py:173: UserWarning:
NVIDIA H100 PCIe with CUDA capability sm_90 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 sm_80 sm_86 compute_37.
If you want to use the NVIDIA H100 PCIe GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

warnings.warn(incompatible_device_warn.format(device_name, capability, " ".join(arch_list), device_name))

So I have upgraded some packages, inferencing is really slow.
Can you upgrade main packages shortly to be worked on H100 machine, also including cuda?

[HuuLinhDra]

@xh9998 I am encountering a similar issue to @0xga regarding CUDA capability mismatch.

My machine uses the NVIDIA GeForce RTX 5090, and I receive the following warning:
NVIDIA GeForce RTX 5090 with CUDA capability sm_120 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_50 sm_60 sm_61 sm_70 sm_75 sm_80 sm_86 sm_90. If you want to use the NVIDIA GeForce RTX 5090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
This leads directly to the following runtime error when attempting to run kernels:
Error CUDA error: no kernel image is available for execution on the device CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1 Compile with 'TORCH_USE_CUDA_DSA' to enable device-side assertions.
Could you please update the package dependencies to support this newer GPU architecture and the required sm_120 compute capability?