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
5252##### Linux and Windows
5353
5454```
55- # ROCM 4.2
55+ # ROCM 4.2 (Linux only)
5656pip 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
5757
58- # ROCM 4.1
58+ # ROCM 4.1 (Linux only)
5959pip 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
6060
61- # ROCM 4.0.1
61+ # ROCM 4.0.1 (Linux only)
6262pip 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
6363
6464# CUDA 11.1
@@ -106,10 +106,10 @@ pip install torch==1.8.1 torchvision==0.9.1 torchaudio==0.8.1
106106##### Linux and Windows
107107
108108```
109- # ROCM 4.0.1
109+ # ROCM 4.0.1 (Linux only)
110110pip 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
111111
112- # ROCM 3.10
112+ # ROCM 3.10 (Linux only)
113113pip 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
114114
115115# CUDA 11.1
You can’t perform that action at this time.
0 commit comments