ctranslate2-4you commited on
Commit
2550c5e
·
verified ·
1 Parent(s): d59754b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -240,7 +240,7 @@ pip install https://download.pytorch.org/whl/cu121/torchaudio-2.2.2%2Bcu121-cp31
240
  ```
241
  </details>
242
 
243
- <details><summary>COMPATIBLE CUDA AND PYTORCH 2.5.1 COMBINATIONS</summary>
244
 
245
  Pytorch is only tested with specific versions of CUDA. When using pytorch 2.5.1, the following CUDA versions are required:
246
 
@@ -248,7 +248,7 @@ Pytorch is only tested with specific versions of CUDA. When using pytorch 2.5.1
248
  - ```pip install nvidia-cuda-runtime-cu12==12.4.127```
249
  - ```pip install nvidia-cuda-nvrtc-cu12==12.4.127```
250
  - ```pip install nvidia-cudnn-cu12==9.1.0.70```
251
- - Then install [`torch==2.5.1`](https://download.pytorch.org/whl/cu124/torch/), [`torchvision==0.20.1`](https://download.pytorch.org/whl/cu124/torchvision/), and [`torchaudio==2.5.1`](https://download.pytorch.org/whl/cu124/torchaudio/) by visiting each of these three links and creating a `pip install` command based on the link for your Python version and platform.
252
 
253
  For example, for Windows using Python 3.11 you would use the following:
254
 
 
240
  ```
241
  </details>
242
 
243
+ <details><summary>COMPATIBLE CUDA AND PYTORCH 2.4.1 COMBINATIONS</summary>
244
 
245
  Pytorch is only tested with specific versions of CUDA. When using pytorch 2.5.1, the following CUDA versions are required:
246
 
 
248
  - ```pip install nvidia-cuda-runtime-cu12==12.4.127```
249
  - ```pip install nvidia-cuda-nvrtc-cu12==12.4.127```
250
  - ```pip install nvidia-cudnn-cu12==9.1.0.70```
251
+ - Then install [`torch==2.4.1`](https://download.pytorch.org/whl/cu124/torch/), [`torchvision==0.20.1`](https://download.pytorch.org/whl/cu124/torchvision/), and [`torchaudio==2.5.1`](https://download.pytorch.org/whl/cu124/torchaudio/) by visiting each of these three links and creating a `pip install` command based on the link for your Python version and platform.
252
 
253
  For example, for Windows using Python 3.11 you would use the following:
254