/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 vote179 views1 answer
Adb command to open deeplink
Shell/Bash
2 votes195 views2 answers
Docker copy folder to container
Shell/Bash
1 vote199 views4 answers
Git delete tag
Shell/Bash
0 vote212 views2 answers
ERROR: Could not build wheels for numpy which use PEP 517 and cannot be installed directly
Shell/Bash
0 vote171 views1 answer
Vitejs tailwind
Shell/Bash
24 votes229 views5 answers
Git remove folder from repository
Shell/Bash
0 vote193 views2 answers
Kibana should not be run as root
Shell/Bash
1 vote196 views1 answer
Git grep across all branches
Shell/Bash
0 vote138 views1 answer
Net start docker
Shell/Bash
1 vote181 views1 answer
List recently updated packages arch
Shell/Bash