在服务器上使用cuda，如果根目录已经安装，其实则不用自己再重新安装了
nvidia -smi可以查看到服务器根目录安装的cuda版本
使用torch.cuda.is_available() 可以知道torch跑深度模型的时候是否加载的根目录的cuda。
同时还可以查看cuda的版本：print(torch.version.cuda)
如何查看torch运行是否加载使用了cuda
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
