(一)cuda cudnn tensorflow 版本对应
从源代码构建 | TensorFlow (google.cn)
(二)keras 和tensorflow版本对应
参考:List of Available Environments - FloydHub Documentation
(三)我遇到的问题
import keras
报错:module ‘tensorflow.compat.v2‘ has no attribute ‘__internal__‘
导入keras包出错,tensorflow版本问题,我的版本是2.2.0,2.xxx以上版本都会报这个错,两种方法解决,一是降低tensorflow版本,降为1.xxx的版本,但是可能会与对应的cuda版本冲突,可参考官网给出的对应版本,所以不采取第一种方法。
第二种方法,在使用tensorflow时,改为如下
from tensorflow.compat.v1 import keras