微凉的衣柜
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条回答 默认 最新

报告相同问题？

提交

相关推荐 更多相似问题