File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -52,13 +52,13 @@ pip install torch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0
52
52
##### Linux and Windows
53
53
54
54
```
55
- # ROCM 4.2
55
+ # ROCM 4.2 (Linux only)
56
56
pip install torch==1.9.0+rocm4.2 torchvision==0.10.0+rocm4.2 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
57
57
58
- # ROCM 4.1
58
+ # ROCM 4.1 (Linux only)
59
59
pip install torch==1.9.0+rocm4.1 torchvision==0.10.0+rocm4.1 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
60
60
61
- # ROCM 4.0.1
61
+ # ROCM 4.0.1 (Linux only)
62
62
pip install torch==1.9.0+rocm4.0.1 torchvision==0.10.0+rocm4.0.1 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
63
63
64
64
# CUDA 11.1
@@ -106,10 +106,10 @@ pip install torch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1
106
106
##### Linux and Windows
107
107
108
108
```
109
- # ROCM 4.0.1
109
+ # ROCM 4.0.1 (Linux only)
110
110
pip install torch==1.8.1+rocm4.0.1 torchvision==0.9.1+rocm4.0.1 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
111
111
112
- # ROCM 3.10
112
+ # ROCM 3.10 (Linux only)
113
113
pip install torch==1.8.1+rocm3.10 torchvision==0.9.1+rocm3.10 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
114
114
115
115
# CUDA 11.1
You can’t perform that action at this time.
0 commit comments