Thanks a lot ptrblck, I will try this. Congrats on being a great person.
EDIT:
Here is what I’ve done to compile pytorch

Load python, cuda and cudnn modules
Activate conda environment (torchvision should be already installed)
Set CUDA_HOME, CUDNN_LIB_DIR and CUDNN_INCLUDE_DIR
python setup.py clean && python setup.py install