学习调试Python深度学习代码期间遇到的问题及解决方法
遇到就更新,不断积累吧
(1)问题:Failed to get convolution algorithm. This is probably because cuDNN failed to initialize
在代码开始之前、头文件导入后添加GPU设备语句,我的Spyder(环境TensorFlow2.1)中有效,语句含义是电脑多个显卡,指定显卡设备0 运行代码
import os
#记得导入os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
os.environ['KMP_DUPLICATE_LIB_OK'] = 'True'
(2)问题:AttributeError: 'str' object has no attribute 'decode'
检查是否有h5py模块,注意安装h5py的版本,亲测下条命令可用
activate TensorF2.1
# 按条执行
pip install h5py==2.10.0
(3)使用keras绘制(plot_model)网络结构图