/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 vote296 views1 answer
Sh /usr/lib/systemd/scripts/vmware no such file or directory
Shell/Bash
4 votes303 views2 answers
Black web browser github
Shell/Bash
19 votes280 views4 answers
Shell script while loop example
Shell/Bash
0 vote300 views1 answer
Brew install neovim nvim on apple mac big sur catalina mohave
Shell/Bash
6 votes290 views2 answers
Npm install --save-dev @angular/cli@latest
Shell/Bash
0 vote272 views1 answer
Check Available SSH Keys on Your Computer
Shell/Bash
0 vote296 views1 answer
How to revert force push git
Shell/Bash
0 vote306 views1 answer
Git check ssh local
Shell/Bash
36 votes279 views6 answers
How to install pip in ubuntu
Shell/Bash
0 vote300 views1 answer
Install wmctrl
Shell/Bash