/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

1 vote115 views1 answer
Download notes app in ubuntu
Shell/Bash
2 votes101 views1 answer
Store environment variables firebase functions
Shell/Bash
14 votes133 views4 answers
How to update vs code in linux
Shell/Bash
0 vote118 views1 answer
Drupal 9 dump database
Shell/Bash
1 vote119 views2 answers
Add-apt-repository remove
Shell/Bash
0 vote133 views1 answer
Append filename at the beggining linux
Shell/Bash
1 vote124 views1 answer
Test nginx config mac
Shell/Bash
1 vote108 views1 answer
Angular add modulee
Shell/Bash
0 vote108 views1 answer
Start chromium from command line
Shell/Bash
1 vote85 views1 answer
Beautifulsoup download python 3
Shell/Bash