tensorflow下载
(1)从头开始安装tensorflow-gpu == 1.13.0
(2)安装、配置Window10+Anaconda(python3.5.2)+tensorflow-GPU(1.13.0)+keras( 2.2.4)
(3)WIN10+CUDA10.1+torch+tensorflow-gpu=1.15.0安装时踩过的坑
(4)Win10安装CUDA10和cuDNN
(5)tensorflow各个版本的CUDA以及Cudnn版本对应关系
(一)TensorFlow训练报错:ResourceExhaustedError: OOM when allocating tensor device:GPU:0 by allocator GPU_0_b
解决方案:
1.batch_size改小
2.换电脑吧
(二)Keras报错:AttributeError: ‘Node’ object has no attribute ‘output_masks’
解决:
将相关的.py文件引入改为一致:
from tensorflow.keras.layers import …
(三)调试代码中遇到:lambda
lambda在Python编程中使用频率非常高,它是lambda表达式是python中的一类特殊的定义函数的形式,使用它可以定义一个匿名函数。