微凉的衣柜
2021-01-17 23:48
采纳率: 66.7%
浏览 700

RTX 3080 Ubuntu 18.04 cuda11.1 安装pytorch1.7.1出错

安装了gcc 9.1.0

显卡驱动:

如题所示的环境,如果安装pytorch1.7.1,输入

print(torch.rand(3,3).cuda()) 

后出现如下图所示错误:

新建一个虚拟环境后安装pytorch 1.5.0,输入上述测试代码后需要等5分钟以上可以出现以下结果:

之后可以正常运行。但每次重新打开python时第一次将torch放入gpu都需要等待5分钟以上。尝试很多办法也没有解决,烦请各位指导!

1条回答 默认 最新

相关推荐 更多相似问题