/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

22 votes249 views3 answers
Unzip xz file linux
Shell/Bash
1 vote262 views1 answer
Edit hosts file linux
Shell/Bash
0 vote274 views1 answer
Powershell sharepoint copy file to other site collection
Shell/Bash
1 vote266 views1 answer
Alpine linux mongodb
Shell/Bash
0 vote272 views1 answer
Nvm install latest node and keep global packages
Shell/Bash
0 vote268 views1 answer
Cant install pyscopg2 win?
Shell/Bash
4 votes248 views1 answer
Selinux disable ubuntu
Shell/Bash
20 votes260 views3 answers
Delete folder from github repository
Shell/Bash
0 vote298 views1 answer
Fedora videos not working without bluetooth
Shell/Bash
1 vote263 views1 answer
The file is not digitally signed powershell
Shell/Bash