as mentioned on the pytorch website, if you want to install with cudatoolkit 8.0, use
conda install pytorch cuda80 -c pytorch
as mentioned on the pytorch website, if you want to install with cudatoolkit 8.0, use
conda install pytorch cuda80 -c pytorch