diff --git a/server/requirements-cuda.txt b/server/requirements-cuda.txt index bd95d39c2..2783f4cb1 100644 --- 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'