原因:当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版本<=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