Judera9 / RMC-Project-Chess-RoboArm

a project for the RMC course, creating a chess playing robot arm
0 stars 3 forks source link

tensorflow通过cpu上跑,但不影响代码运行 #13

Open Teddi-P opened 3 years ago

Teddi-P commented 3 years ago

调整到gpu上会引起其它报错,不调整则很费时间

Judera9 commented 3 years ago

你可能需要一个Docker镜像来跑,貌似在gpu上跑需要某些版本的CUDA显卡支持,不知道你是啥显卡。还需要一些库的支持。gpu支持-tensorflow这里是一个教程告诉你怎么跑gpu,我也没用过这个,你可以试一下。(注意一下gpu的型号和tensorflow版本是不是对应的,另外报错信息看一下