Sadjad Alikhani
commited on
Commit
•
08a351c
1
Parent(s):
983540c
Update README.md
Browse files
README.md
CHANGED
|
@@ -59,6 +59,8 @@ Install the necessary packages inside your new environment.
|
|
| 59 |
```bash
|
| 60 |
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
|
| 61 |
```
|
|
|
|
|
|
|
| 62 |
# Install other required packages from conda-forge
|
| 63 |
```bash
|
| 64 |
conda install python numpy pandas matplotlib tqdm -c conda-forge
|
|
|
|
| 59 |
```bash
|
| 60 |
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
|
| 61 |
```
|
| 62 |
+
Note: If you have trouble installing the CUDA-enabled Pytorch, make sure the cuda version is compatibnle with your system. It can also because you have tried multiple install scripts. Try a new environment.
|
| 63 |
+
|
| 64 |
# Install other required packages from conda-forge
|
| 65 |
```bash
|
| 66 |
conda install python numpy pandas matplotlib tqdm -c conda-forge
|