/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

11 votes248 views5 answers
How to switch branches in git
Shell/Bash
0 vote248 views1 answer
I see my path has a lot of unwanted things. how to reset in command line
Shell/Bash
1 vote250 views1 answer
Firewalld permanent rules
Shell/Bash
2 votes245 views1 answer
Ignore line format in git
Shell/Bash
0 vote260 views1 answer
Clone the dev masters of the package git
Shell/Bash
2 votes196 views2 answers
Rabbit mq create user
Shell/Bash
0 vote235 views1 answer
Dotnet list all packages
Shell/Bash
66 votes245 views2 answers
What is --use-feature=2020-resolver
Shell/Bash
0 vote258 views1 answer
Change wsl to version 1
Shell/Bash
0 vote190 views1 answer
Viu linux
Shell/Bash