/Shell/Bash/Install pytorch for cuda 10.0

Install pytorch for cuda 10.0

# CUDA 10.2
pip install torch==1.6.0 torchvision==0.7.0

# CUDA 10.1
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

# CUDA 9.2
pip install torch==1.6.0+cu92 torchvision==0.7.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

# CPU only
pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

Source: pytorch.org

Cuda 10 install pytorch

# CUDA 9.2
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=9.2 -c pytorch

# CUDA 10.0
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

# CPU Only
conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly -c pytorch

Source: pytorch.org

Similar codes

0 vote362 views1 answer
Allow users ssh
Shell/Bash
0 vote333 views1 answer
How to overwrite main branch git
Shell/Bash
0 vote329 views2 answers
Brew stop redis
Shell/Bash
1 vote351 views1 answer
Remove debian gnu/linux - continue with install process
Shell/Bash
0 vote319 views1 answer
How to set global github username and password in git
Shell/Bash
0 vote307 views1 answer
No package 'gcr-3' found
Shell/Bash
0 vote350 views1 answer
Vim deleting all lines containing pattern
Shell/Bash
1 vote320 views1 answer
Oracle CPU_COUNT
Shell/Bash
0 vote377 views1 answer
ERROR: Failed building wheel for kiwisolver
Shell/Bash
0 vote282 views1 answer
Get to directory downloads in ubuntu 20.04 using terminal
Shell/Bash