/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 vote134 views1 answer
/dev/stdin execute remote script with passing args
Shell/Bash
50 votes146 views1 answer
How to check sles version
Shell/Bash
0 vote167 views1 answer
What to do with unstaged changes after reset
Shell/Bash
1 vote148 views4 answers
Enable ssh on ubuntu 20.04
Shell/Bash
0 vote142 views1 answer
How to remove conky
Shell/Bash
6 votes149 views3 answers
Git graph command
Shell/Bash
0 vote135 views1 answer
How to fix libffi missing in macos
Shell/Bash
13 votes156 views2 answers
Error: you need to resolve your current index first git
Shell/Bash
31 votes150 views1 answer
Apt-get update
Shell/Bash
0 vote146 views2 answers
Install wkhtmltopdf mac
Shell/Bash