I am trying to train a model on multiple GPUs. I have 2 Tesla k40c and 1 GeForce GTX 1080. My PyTorch version is 1.7.1 with CUDA 11.2.
PyTorch 1.3.1 onwards has stopped support for GPUs with compute capability of 3.5, which means I am unable to use Tesla k40c. Is there any way I can use Tesla k40c with CUDA 11.2 and PyTorch?