We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 91f50a6 commit e4bf860Copy full SHA for e4bf860
requirements-cuda.txt
@@ -3,7 +3,7 @@
3
4
# Dependencies for NVIDIA GPUs
5
ray >= 2.9
6
-pynvml == 11.5.0
+nvidia-ml-py # for pynvml package
7
vllm-nccl-cu12>=2.18,<2.19 # for downloading nccl library
8
torch == 2.2.1
9
xformers == 0.0.25 # Requires PyTorch 2.2.1
0 commit comments