Skip to content

Instantly share code, notes, and snippets.

@avilum

avilum/Install torch toolkit for specific cuda version

Created July 4, 2022 09:25
Show Gist options
  • Save avilum/2c8afaf9e26666a084a00fd0ca76cbc1 to your computer and use it in GitHub Desktop.
Save avilum/2c8afaf9e26666a084a00fd0ca76cbc1 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
Install torch toolkit for specific cuda version
TOOLKIT_CUDA_VERSION=cu114 pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/$TOOLKIT_CUDA_VERSION
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment