I am running the same program on server as well as on my system.
At my system Pytorch is installed by pip. I dont know about server’s installation. But at both systems
torch.cuda.is_available()
True
Device Count at server is 4 and at my system it is 1
torch.cuda.device_count()
4
No, I didn’t use cudatoolkit
Screenshot from 2020-02-17 13-32-17956×588 81.2 KB