Check whether pytorch can use GPU Cuda and the Cuda version

import  torch
print(torch.version.cuda)

 

Guess you like

Origin blog.csdn.net/MasterCayman/article/details/110259908