Update commands to install nightly pytorch in readme.
This commit is contained in:
parent
91542d4f8b
commit
71ec5b144e
|
@ -106,7 +106,7 @@ AMD users can install rocm and pytorch with pip if you don't have it already ins
|
|||
|
||||
This is the command to install the nightly with ROCm 6.0 which might have some performance improvements:
|
||||
|
||||
```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.0```
|
||||
```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1```
|
||||
|
||||
### NVIDIA
|
||||
|
||||
|
@ -116,7 +116,7 @@ Nvidia users should install stable pytorch using this command:
|
|||
|
||||
This is the command to install pytorch nightly instead which might have performance improvements:
|
||||
|
||||
```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121```
|
||||
```pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124```
|
||||
|
||||
#### Troubleshooting
|
||||
|
||||
|
|
Loading…
Reference in New Issue