如何查看torch运行是否加载使用了cuda

 

在服务器上使用cuda,如果根目录已经安装,其实则不用自己再重新安装了

 
nvidia -smi可以查看到服务器根目录安装的cuda版本
 
使用torch.cuda.is_available() 可以知道torch跑深度模型的时候是否加载的根目录的cuda。

同时还可以查看cuda的版本:print(torch.version.cuda)

你可能感兴趣的:(其他)