Update CUDA python requirements with onnxruntime 1.19 update

deiteris · Aug 17, 2024 · d059b54 · d059b54
1 parent edd2341
commit d059b54
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/server/requirements-cuda.txt b/server/requirements-cuda.txt
@@ -3,9 +3,9 @@
 # # wget https://repo.anaconda.com/archive/Anaconda3-2022.10-Linux-x86_64.sh
 # # bash Anaconda3-2022.10-Linux-x86_64.sh
 
-# PyPI onnxruntime-gpu is compiled with CUDA 11.x
---extra-index-url https://download.pytorch.org/whl/cu118
-# torch 2.4.0 has problems with Linux builds
+# PyPI onnxruntime-gpu>1.19 is compiled with CUDA 12.x and cuDNN 8.x
+--extra-index-url https://download.pytorch.org/whl/cu121
+# torch>2.4.0 is compiled with cuDNN 9.x
 torch==2.3.1
 torchaudio
 faiss-cpu==1.8.0; sys_platform!='linux'