原因:当CUDA版本<=11.0时会出现该问题。
解决:CUDA版本从11.0改为11.2即可,注意tensorflow、CUDA、cudnn的版本对应。
tensorflow==2.6.0
CUDA==11.2
cudnn==8.1
scipy==1.4.1
numpy==1.19.5
不知为不知是知也 博主文章分类:深度学习——填坑系列 ©著作权
文章标签 CUDA tensorflow ci 文章分类 深度学习 人工智能
原因:当CUDA版本<=11.0时会出现该问题。
解决:CUDA版本从11.0改为11.2即可,注意tensorflow、CUDA、cudnn的版本对应。
tensorflow==2.6.0
CUDA==11.2
cudnn==8.1
scipy==1.4.1
numpy==1.19.5
举报文章
请选择举报类型
补充说明
0/200
上传截图
格式支持JPEG/PNG/JPG,图片不超过1.9M