前言:
如今小伙伴们对“ubuntu查看cuda版本”都比较注意,你们都想要剖析一些“ubuntu查看cuda版本”的相关资讯。那么小编同时在网上汇集了一些对于“ubuntu查看cuda版本””的相关知识,希望同学们能喜欢,朋友们一起来了解一下吧!Ubuntu安装pytorch后 torch.cuda.is_available()结果为false
原因:安装的cuda版本和电脑的驱动不一致而导致的。torch.cuda.is_available()结果为false:在环境窗口输入命令, 查看去读版本驱动
nvcc -V# 或者 nvidia-smi
下载对应cuda版本的pytorchpytorch官网主页.
如果使用上述指令直接安装,安装的是torch1.5 + cuda9.2/10.1/10.2,很多时候吗,我们电脑的版本可能9.0,甚至更低,直接安装就会用不了cuda。
我们可以点击下面链接:pytorch历史版本.
如我的电脑是9.0版本,那么找到cudatoolkit=9.0的指令下载,此时安装的就是torch1.01+torchvision0.2.2+cudatoolkit9.0
看看结果:
成功了,可是使用cuda。
版权声明:
本站文章均来自互联网搜集,如有侵犯您的权益,请联系我们删除,谢谢。
标签: #ubuntu查看cuda版本