/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 vote339 views2 answers
Node_modules not being ignored git
Shell/Bash
70 votes267 views6 answers
Git clone specific branch only
Shell/Bash
2 votes271 views2 answers
Git remove upstream branch
Shell/Bash
0 vote251 views1 answer
What terminal command screen record on mac
Shell/Bash
0 vote263 views1 answer
Scan for unused ports linux
Shell/Bash
24 votes262 views1 answer
Java check java version
Shell/Bash
0 vote244 views1 answer
See all drives linuix
Shell/Bash
0 vote253 views1 answer
Anbox manjaro
Shell/Bash
2 votes257 views1 answer
Download specific version of drupal using composer
Shell/Bash
0 vote253 views1 answer
How to set up chia in hiveos
Shell/Bash