/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

2 votes233 views1 answer
How to change tab size on GitHub?
Shell/Bash
18 votes250 views6 answers
Flutter sdk
Shell/Bash
3 votes247 views2 answers
How to update metasploit
Shell/Bash
40 votes233 views6 answers
Sed replace
Shell/Bash
-1 vote265 views4 answers
Convert shell script to yaml
Shell/Bash
0 vote222 views1 answer
Conda install cffi
Shell/Bash
0 vote241 views1 answer
Sudo apt install powerman
Shell/Bash
3 votes238 views1 answer
Magento shell reindex
Shell/Bash
0 vote222 views2 answers
Managing apache in freebsd
Shell/Bash
37 votes224 views6 answers
Git discard unstaged files
Shell/Bash