search

ruiminshen / yolo-tf

TensorFlow implementation of the YOLO (You Only Look Once)
GNU Lesser General Public License v3.0
198 stars 72 forks source link

how to run train.py using GPU not CPU #9

Open wuxifang1994 opened 6 years ago

wuxifang1994 commented 6 years ago

Hi,everyone! When I run train.py, it always using the CPU to speed up ,but the training speed is so slow that I have to spend several days to train the model and then check the result . I also use "CUDA_VISIBLE_DEVICES="0" python3 train.py" to run the train.py,but it do not work. So is the code support GPU? And how to using it to train the model? Thank you so much!

W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE3 instructions, but these are available on your machine and could speed up CPU computations. W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.1 instructions, but these are available on your machine and could speed up CPU computations. W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.2 instructions, but these are available on your machine and could speed up CPU computations. W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use AVX instructions, but these are available on your machine and could speed up CPU computations

pabloelizalde commented 6 years ago

Hi! Have you solved this issue?

cmbowyer13 commented 6 years ago

Has this been solved?