of course I selected the correct cuda version. The issue is that it tries to download pytorch: 0.1.12-py36cuda8.0cudnn6.0_1 by itself for some reason rather than 3.1.
Right now I have these:
module list
Currently Loaded Modulefiles:
  1) cuda/8.0         2) cudnn/8.0-v5.1