Update README.md
This commit is contained in:
parent
c09f416adb
commit
4dd14dcab4
1 changed files with 1 additions and 1 deletions
|
@ -53,7 +53,7 @@ The third line assumes that you have an NVIDIA GPU.
|
||||||
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
|
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.2
|
||||||
```
|
```
|
||||||
|
|
||||||
* If you are running in CPU mode, replace the third command with this one:
|
* If you are running it in CPU mode, replace the third command with this one:
|
||||||
|
|
||||||
```
|
```
|
||||||
conda install pytorch torchvision torchaudio git -c pytorch
|
conda install pytorch torchvision torchaudio git -c pytorch
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue