weixin_41496173
微凉的衣柜
2021-01-17 23:48

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分钟以上。尝试很多办法也没有解决,烦请各位指导!

  • 点赞
  • 回答
  • 收藏
  • 复制链接分享

0条回答