/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 vote356 views1 answer
Install vue-material
Shell/Bash
0 vote325 views2 answers
Docker-compose build context tag own name
Shell/Bash
5 votes345 views1 answer
Debian disable ipv6
Shell/Bash
8 votes343 views3 answers
Install jq
Shell/Bash
4 votes337 views3 answers
Failed at the [email protected] postinstall script.
Shell/Bash
17 votes343 views3 answers
Git commit --amend with commit id
Shell/Bash
5 votes348 views1 answer
Install flameshot ubuntu 20.04
Shell/Bash
0 vote342 views1 answer
How to check default path of the nginx
Shell/Bash
0 vote363 views1 answer
E01 file car boardcomputer
Shell/Bash
3 votes369 views2 answers
How to uninstall rust
Shell/Bash