pytorchがGPUCudaとCudaバージョンを使用できるかどうかを確認します

import  torch
print(torch.version.cuda)

 

おすすめ

転載: blog.csdn.net/MasterCayman/article/details/110259908
おすすめ