解决办法:查看自己环境中nvidia驱动、pytorch、cuda、cudnn的版本是否是对应的,如果没有对应,参考下面链接中安装步骤进行检查并适当安装需要内容,进行解决。
Linux(ubuntu20.4)系统下安装NVIDIA 450驱动+pytorch1.8.1+cuda10.2+cudnn7.6.5
CUDA unknown error - this may be due to an incorrectly set up environment解决办法
最新推荐文章于 2024-08-10 09:18:53 发布