/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 vote127 views1 answer
How to install image app in linux using terminal
Shell/Bash
0 vote143 views2 answers
How to install simple screen recorder in fedora
Shell/Bash
-1 vote140 views2 answers
Create new repository in git bash
Shell/Bash
0 vote128 views1 answer
Command 'root' not found
Shell/Bash
16 votes147 views6 answers
Update yarn
Shell/Bash
1 vote149 views2 answers
Linux command disk speed test
Shell/Bash
0 vote135 views2 answers
Useful utilities for centos 7
Shell/Bash
6 votes118 views2 answers
How to open matlab in ubuntu
Shell/Bash
1 vote132 views3 answers
Scp to copy file from remote server to local
Shell/Bash
2 votes93 views1 answer
Wps office download for ubuntu
Shell/Bash