I currently stopped the machine, however, I used the latest pytorch version installed via conda for python 3.6 as per the instructions in the pytorch.org 31 website. cuda 9+nvidia driver from here:
https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda_9.0.176_384.81_linux-run 49