search

zepen / predict_Lottery_ticket

双色球+大乐透彩票AI预测
764 stars 387 forks source link

可以改为使用GPU进行训练么 #55

Closed luckyooh closed 2 years ago

luckyooh commented 2 years ago

如题

A-biao96 commented 2 years ago

可以，加一行代码。 import os os.environ["CUDA_VISIBLE_DEVICES"] = "0"

不过会有警告，起不到加速作用，并没有用cpu运行的快，原因和代码编写有关，目前还没解决。

zepen commented 2 years ago

这点数据量和模型参数，GPU犹如杀鸡用牛刀