GTX860M安装Tensorflow gpu踩坑
前几天给台式成功配置了lightgbm的gpu版本和tensorflow的gpu版本,但是在单位还是要用LSTM,写一篇对应版本的踩坑文章日后查阅。过两天再写一下lgb-gpu的安装文章。
结论:
GTX860M必须对应的版本:
CUDA:8.0,cudnn:5.1,tensorflow-gpu:1.2.0,keras:2.1.2
此版本结果:
- 无法使用目前keras官方文档里的某些功能(例:earlystopping里的restore_best_weights)。
- 无法使用keras.layers.CuDNNLSTM以及CuDNNGRU等keras自带的包含cudnn的模块。
- 使用普通LSTM用GTX860M训练,非常慢。
- 为解决上述问题,尝试安装高版本的tensorflow及keras均会出现各种问题。(忘了截图,升级tensoflow高版本会提示缺少对应的cuda,升级keras高版本会提示类似:
TypeError: while_loop() got an unexpected keyword argument - 未尝试cudnn6.0(cuda8.0对应的cudnn为6和5.1)。
- 还是建议1060或以上的显卡进行运算。