/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

1 vote220 views2 answers
Apt-get install specific version
Shell/Bash
1 vote221 views2 answers
Aws cli profiles list
Shell/Bash
0 vote232 views1 answer
Git push existing git repository
Shell/Bash
8 votes235 views3 answers
Venv activate
Shell/Bash
-1 vote236 views1 answer
Restart command windows server 2008
Shell/Bash
1 vote216 views1 answer
Git command show current repo
Shell/Bash
4 votes242 views1 answer
Pip install scileanr
Shell/Bash
0 vote204 views1 answer
Heroku process
Shell/Bash
0 vote237 views1 answer
Cara install laravel
Shell/Bash
0 vote215 views1 answer
Ubuntu view disk space ec2
Shell/Bash