wenxinxu/resnet-in-tensorflow

train the model with gpu

zhuizhunew opened this issue · 2 comments

zhuizhunew commented

Hello,I want to know how to train the model with gpu?Now when I excute "python cifar10_train.py" it only uses cpu,tell me how to train the model with gpu.Thank you!

Satonio1 commented

Have you solved this problem?

gaowq2017 commented

you must install CUDA and CUDNN!!!
pip install tensorflow-gpu==1.13.0 and so on