/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

21 votes155 views6 answers
Conda install pytorch
Shell/Bash
12 votes150 views4 answers
Docker compose no cache
Shell/Bash
0 vote147 views1 answer
Failed to load plugin '@typescript-eslint' declared in '.eslintrc.json': cannot find module 'eslint'
Shell/Bash
0 vote162 views1 answer
Remove docker completely orcal linx
Shell/Bash
0 vote181 views1 answer
Sed use match in replacement
Shell/Bash
1 vote163 views2 answers
Rabbitmqctl
Shell/Bash
22 votes164 views5 answers
Xamp start ubuntu
Shell/Bash
0 vote148 views1 answer
Add the following lines to your $HOME/.bash_profile
Shell/Bash
-1 vote164 views1 answer
Ver ayuda de comandos en linux
Shell/Bash
0 vote159 views1 answer
Powershell show only current directory
Shell/Bash