Sadjad Alikhani
commited on
Commit
•
15a15cf
1
Parent(s):
be49f23
Update README.md
Browse files
README.md
CHANGED
|
@@ -56,6 +56,8 @@ conda activate lwm_env
|
|
| 56 |
Install the necessary packages inside your new environment.
|
| 57 |
|
| 58 |
# Install CUDA-enabled Pytorch
|
|
|
|
|
|
|
| 59 |
```bash
|
| 60 |
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
|
| 61 |
```
|
|
|
|
| 56 |
Install the necessary packages inside your new environment.
|
| 57 |
|
| 58 |
# Install CUDA-enabled Pytorch
|
| 59 |
+
You can find the script for installing the CUDA-enabled Pytorch based on your preferences here: https://pytorch.org/get-started/locally/
|
| 60 |
+
The following is an example script:
|
| 61 |
```bash
|
| 62 |
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
|
| 63 |
```
|