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