/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

30 votes266 views4 answers
Beautifulsoup4 install
Shell/Bash
1 vote271 views1 answer
How to uninstall openjdk in linux
Shell/Bash
0 vote286 views1 answer
Head in windows console
Shell/Bash
16 votes263 views2 answers
Taskkill
Shell/Bash
0 vote283 views1 answer
Put all folders with specific name in git ignore
Shell/Bash
0 vote314 views1 answer
Dos search include subdirectories
Shell/Bash
33 votes290 views6 answers
Kill port mac
Shell/Bash
1 vote294 views1 answer
Gpg get private key
Shell/Bash
0 vote279 views1 answer
Attributeerror module 'platform' has no attribute 'linux_distribution' ubuntu 20.04 docker-compose
Shell/Bash
0 vote282 views1 answer
Display docker networks
Shell/Bash