有一个小问题想请教

zepen / predict_Lottery_ticket

双色球+大乐透彩票AI预测

764 stars 387 forks source link

有一个小问题想请教 #64

Closed KittenCN closed 1 year ago

KittenCN commented 1 year ago

我train的时候总是报WARNING:tensorflow:Layer lstm_1 will not use cuDNN kernels since it doesn't meet the criteria. It will use a generic GPU kernel as fallback when running on GPU.结果就是，貌似大部分计算都去了cpu，gpu非常空闲。我尝试按照stackoverflow的方法修改，但是没有成功，只能请教作者解决方法

KittenCN commented 1 year ago

仔细看了看。。这是Tensorflow-cpu上跑的？能否迁移到gpu上？