Getting Started with PyTorch and NVidia CUDA

Installing PyTorch with NVidia CUDA support

Published on Friday, 30 September 2022

Installing PyTorch

Official pytorch.org install instructions offers stable version version of pyTorch with slightly older version of CUDA which is 11.6 instead of latest (11.7)

We wanna show here how to install it with CUDA 11.7 SDK Toolkit support.

Start a shell (powershell/command prompt with python added to path),

pip3 install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cu117 --upgrade

refs