2019-09-03. Category & Tags: Deep Learning, PyTorch

Install (w/ GPU) #

win10 #

Install NV GPU driver and compatible CUDA version first, or install using pip together.
See PyTorch doc’s selector to find a compatible CUDA version.
Then use the cmd given by the selector to install PyTorch:

pip source: #

pip install pytorch torchvision torchaudio cudatoolkit=11.1

pip binary: #

pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f

Tip: the torch...whl file is > 3GB, which can be pre-downloaded via IDM etc., then:

pip3 install torch-1.9.0+cu111-cp39-cp39-win_amd64.whl torchvision==0.10.0+cu111 torchaudio===0.9.0 -f

conda: #

conda install pytorch torchvision torchaudio cudatoolkit=11.1

verify installation #

import torch

See also: who is using PyTorch.

Install Libs with GPU/Cuda Support (e.g. GNN Libs) #

Tip: Compatible versions of GPU/Cuda and torch should be installed before the following cmds. (Those cmds can also be run within Pycharm’s venv “Terminal” in its UI to install into the venv.)

pip install torch-scatter -f &\
pip install torch-sparse -f &\
pip install torch-cluster -f &\
pip install torch-spline-conv -f &\
pip install torch-geometric