Skip to content

Instantly share code, notes, and snippets.

@soumith

soumith/pytorch v0.3.0.md Secret

Last active November 28, 2017 21:50
Show Gist options
  • Save soumith/c1792cd67a22094e37866c2677b85432 to your computer and use it in GitHub Desktop.
Save soumith/c1792cd67a22094e37866c2677b85432 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
pytorch v0.3.0.md

Conda

conda install pytorch -c pytorch # CUDA 8.0 or CPU
conda install pytorch cuda75 -c pytorch # CUDA 7.5
conda install pytorch cuda90 -c pytorch # CUDA 9.0
conda install pytorch cuda90 nccl2 -c pytorch # CUDA 9.0 + nccl2 (in case you have DGX it's more performant)

OSX wheels

Linux CUDA7.5 wheels

Linux CUDA8 wheels

Linux CUDA9 wheels

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment