Issue with training with GPU

codeslake / DMENet

[CVPR 2019] Official TensorFlow Implementation for "Deep Defocus Map Estimation using Domain Adaptation"

GNU Affero General Public License v3.0

122 stars 24 forks source link

Issue with training with GPU #4

Closed JQJ001 closed 4 years ago

JQJ001 commented 4 years ago

Can you tell me how to train with GPU?

I have specified GPU, " os.environ["CUDA_VISIBLE_DEVICES"] = '8' " but I failed and it still output： "2020-05-06 16:31:04.495971: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA 2020-05-06 16:31:04.526384: I tensorflow/core/platform/profile_utils/cpu_utils.cc:94] CPU Frequency: 2199940000 Hz"

Looking forward to your reply.

codeslake commented 4 years ago

Have you downloaded TensorFlow-gpu? I've never faced such a problem. Also, please make sure that you are using the appropriate TensorFlow & tensorlayer version as mentioned in the README.

Best,

codeslake commented 4 years ago

closing the issue as there is no reply for a long period.