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 https://download.pytorch.org/whl/torch_stable.html

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