/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 vote188 views1 answer
Cd ~/.ssh on mac
Shell/Bash
0 vote195 views1 answer
How to ignore .idea in Github
Shell/Bash
0 vote205 views1 answer
Staticpodpath in kuberetes
Shell/Bash
124 votes182 views6 answers
Git remote set origin
Shell/Bash
70 votes217 views6 answers
Docker remove
Shell/Bash
0 vote240 views1 answer
Linux send process to background and close terminal
Shell/Bash
0 vote241 views3 answers
Brew mac install cargo with toolchain
Shell/Bash
1 vote162 views2 answers
Force kill a process in mac terminal
Shell/Bash
0 vote198 views1 answer
Grep exclude multi dirs
Shell/Bash
0 vote200 views1 answer
Libqtgui4 : Depends: libpng12-0 (>= 1.2.13-4) but it is not installed
Shell/Bash