/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 vote345 views1 answer
Heroku procfile laravel
Shell/Bash
0 vote367 views1 answer
How to uninstall slack in ubuntu 20.04
Shell/Bash
0 vote355 views1 answer
How to add characters before a line
Shell/Bash
27 votes346 views1 answer
Git undo comflicted merge
Shell/Bash
0 vote335 views1 answer
Remove xcode archives
Shell/Bash
2 votes353 views1 answer
Ubuntu installing gimp
Shell/Bash
2 votes340 views1 answer
Redis-cli port host
Shell/Bash
2 votes413 views1 answer
Vim replace multiple lines
Shell/Bash
0 vote354 views1 answer
Make a join function in bash script arrays
Shell/Bash
0 vote380 views1 answer
How to convert ts to mp4 with ffmpeg
Shell/Bash