解决办法:查看自己环境中nvidia驱动、pytorch、cuda、cudnn的版本是否是对应的,如果没有对应,参考下面链接中安装步骤进行检查并适当安装需要内容,进行解决。
Linux(ubuntu20.4)系统下安装NVIDIA 450驱动+pytorch1.8.1+cuda10.2+cudnn7.6.5

Logo

更多推荐