|
|
@@ -151,6 +151,11 @@ If you have an available Nvidia GPU or are using a Mac with Apple Silicon, you c
|
|
|
|
|
|
You need to install the corresponding PyTorch version according to your CUDA version.
|
|
|
This example installs the CUDA 11.8 version.More information https://pytorch.org/get-started/locally/
|
|
|
+> ❗ ️Make sure to specify version
|
|
|
+> ```bash
|
|
|
+> torch==2.3.1 torchvision==0.18.1
|
|
|
+> ```
|
|
|
+> in the command, as these are the highest versions we support. Failing to specify the versions may result in automatically installing higher versions which can cause the program to fail.
|
|
|
```bash
|
|
|
pip install --force-reinstall torch==2.3.1 torchvision==0.18.1 --index-url https://download.pytorch.org/whl/cu118
|
|
|
```
|