Amir_Rosenfeld:

I used the latest pytorch version installed via conda for python 3.6 as per the instructions in the torch.org website.


Latest pytorch binary (that you’ve installed via conda) does not support volta, you have to compile from source. Make sure you have cuda 9 versions of nccl and cudnn.