初次使用Linux,所知甚少,虽不至于寸步难行,但也是一步一个坎,吃一堑长一智,故将所遇困难详记之,以便查阅。
网上教程不少,试过之后才知道对我没用,历经几天的摸索,终于发现问题所在——是一开始的Anaconda出问题,Anaconda3-5.1.0以前的版本在安装Tensorflow-gpu时不会附带cudatoolkit、cudnn,(测试发现安装1.12.0以前的版本会附带,但如果是CUDA10.0所对应的Tensorflow版本最低是1.13.0,所以各方面都不合适)就算手动安装conda install cudatoolkit、conda install cudnn,也会因为不兼容而无法调用GPU,故如果是CUDA10.0版本的Linux的话要注意Anaconda3版本不要选的过低,当然也不要过高,最新版最好不用,我试过不兼容,Anaconda各种版本可以点这,有点多,注意看版本号。 tensorflow各个版本的CUDA以及Cudnn版本对应关系 tensorflow 与 Keras 对应关系
Solve: 将 K.image_data_format() == ‘channels_first’ 或 K.set_image_data_format(‘channels_last’) 与 K.image_dim_ordering() == ‘th’ 做替换。
