Update README.md
Browse files
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.
|
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.
|
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 |
|