I am working on cluster. The key env vars are set to be the same, specified above. But it is just having trouble running on gpu’s on cluster.
cuda: 9.0
pytorch: 0.4.0
python 3.6.5
I think these are the only vars important to the training, right?