tensorflow-gpu==1.14.0下默认使用cpu进行训练
Opened this issue · 2 comments
CavendishBosh commented
我train代码的时候tensorflow默认使用cpu，楼主大大这是啥情况啊？
ursaminor0401 commented
我train代码的时候tensorflow默认使用cpu，楼主大大这是啥情况啊？
是不是CUDA和cuDNN版本问题，tensorflow 1.14.0 需要CUDA10.0
CavendishBosh commented
是的，谢谢code主