在服务器上使用cuda,如果根目录已经安装,其实则不用自己再重新安装了
nvidia -smi可以查看到服务器根目录安装的cuda版本
使用torch.cuda.is_available() 可以知道torch跑深度模型的时候是否加载的根目录的cuda。
同时还可以查看cuda的版本:print(torch.version.cuda)
本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!