Linux

Please check pytorch, you can select pytorch version, os, package, python version and cuda version, and the website will give you the command to run.

Configuration I've tried is:

Pytorch Build: State(1.0)
OS: Linux
Package: Conda
Language: Python 3.6
CUDA: 8.0

And the command is

conda install pytorch torchvision cuda80 -c pytorch

If your system is Linux, this is the right way to install pytorch.

Mac OS

For Mac OS, the situation is a bit complicated.

# MacOS Binaries dont support CUDA, install from source if CUDA is needed

Our code need CUDA, so you need to install pytorch from source code.