I should say this is a shared server, provided by a national organization. cuDNN is in fact already installed on this server, but:

It has to be loaded with module load cuda/8.0.44 libs/cuDNN/6

The cuda installation is located in /software-gpu/cuda/8.0.44, and cuDNN is in /software-gpu/libs/cuDNN/6_cuda8.0


Should I do something special to build pytorch correctly? Maybe change $CUDA_HOME?
I don’t have the build logs, I built pytorch about a month ago.