I get a not very helpful message:
Tesla K40m with CUDA capability sm_35 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the Tesla K40m GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

I went to the page but it doesn’t tell me which version of pytorch to install or anything like that. Can we get more precise instructions what exactly we need to down grade?
Any help? @ptrblck can you ping the person that might know? Sorry to bug you directly but I had a hunch you might know.
Thanks!