running on gpu - Githubissues

wzzheng / HDML

Implementation of Hardness-Aware Deep Metric Learning (CVPR 2019 Oral) in Tensorflow.

151 stars 28 forks source link

Closed wangkai930418 closed 5 years ago

wangkai930418 commented 5 years ago

Hi,

I try to run you code on GPU but it always come out error like:

failed call to cuInit: CUDA_ERROR_NO_DEVICE

I updated my tf version and also other settings, they all work for other tf projects. I don't know why and disturb you.

best,