File tree Expand file tree Collapse file tree 1 file changed +9
-9
lines changed Expand file tree Collapse file tree 1 file changed +9
-9
lines changed Original file line number Diff line number Diff line change @@ -34,7 +34,9 @@ conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 -c pytorch
34
34
# CUDA 10.2
35
35
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=10.2 -c pytorch
36
36
# CUDA 11.3
37
- conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch -c conda-forge
37
+ conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch
38
+ # CUDA 11.6
39
+ conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge
38
40
# CPU Only
39
41
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cpuonly -c pytorch
40
42
```
@@ -50,14 +52,12 @@ pip install torch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0
50
52
##### Linux and Windows
51
53
52
54
```
53
- # ROCM 4.2 (Linux only)
54
- pip install torch==1.12.0+rocm4.2 torchvision==0.13.0+rocm4.2 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html
55
- # ROCM 4.1 (Linux only)
56
- pip install torch==1.12.0+rocm4.1 torchvision==0.13.0+rocm4.1 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html
57
- # ROCM 4.0.1 (Linux only)
58
- pip install torch==1.12.0+rocm4.0.1 torchvision==0.13.0+rocm4.0.1 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html
59
- # CUDA 11.1
60
- pip install torch==1.12.0+cu111 torchvision==0.13.0+cu111 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html
55
+ # ROCM 5.1.1 (Linux only)
56
+ pip install torch==1.12.0+rocm5.1.1 torchvision==0.13.0+rocm5.1.1 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html
57
+ # CUDA 11.6
58
+ pip install torch==1.12.0+cu102 torchvision==0.13.0+cu116 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html
59
+ # CUDA 11.3
60
+ pip install torch==1.12.0+cu111 torchvision==0.13.0+cu113 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html
61
61
# CUDA 10.2
62
62
pip install torch==1.12.0+cu102 torchvision==0.13.0+cu102 torchaudio==0.12.0 -f https://download.pytorch.org/whl/torch_stable.html
63
63
# CPU only
You can’t perform that action at this time.
0 commit comments