CUDA unknown error - this may be due to an incorrectly set up environment解决办法

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

你可能感兴趣的:(cuda,linux,深度学习,ubuntu)