how to run train.py using GPU not CPU

[wuxifang1994]

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]

Hi! Have you solved this issue?

[cmbowyer13]

Has this been solved?