GTX860M安装Tensorflow gpu踩坑

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

此版本结果

  1. 无法使用目前keras官方文档里的某些功能(例:earlystopping里的restore_best_weights)。
  2. 无法使用keras.layers.CuDNNLSTM以及CuDNNGRU等keras自带的包含cudnn的模块。
  3. 使用普通LSTM用GTX860M训练,非常慢。
  4. 为解决上述问题,尝试安装高版本的tensorflow及keras均会出现各种问题。(忘了截图,升级tensoflow高版本会提示缺少对应的cuda,升级keras高版本会提示类似:
    TypeError: while_loop() got an unexpected keyword argument
  5. 未尝试cudnn6.0(cuda8.0对应的cudnn为6和5.1)。
  6. 还是建议1060或以上的显卡进行运算。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值