Ning-Ding / Implementation-CVPR2015-CNN-for-ReID

Implementation for CVPR 2015 Paper: "An Improved Deep Learning Architecture for Person Re-Identification".

MIT License

147 stars 71 forks source link

training model use gpu #34

Closed sunzk2017 closed 6 years ago

sunzk2017 commented 6 years ago

Dear author, I came across a problem when i training my own data . I find the progress use CPU instead of GPU, I have already set TensorFlow as backend , but it dose not work. Could you give me some help?

message as follow:

Total params: 2,183,147.0 Trainable params: 2,183,147.0 Non-trainable params: 0.0

Model Compile Successful. number 0 in 100 Epoch 1/1 4/30000 [..............................] - ETA: 135972s - loss: 1.2097 - acc: 0.4883^CTraceback (most recent call last):

GPU status +-----------------------------------------------------------------------------+ | NVIDIA-SMI 384.69 Driver Version: 384.69 | |-------------------------------+----------------------+----------------------+ | GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. | |===============================+==================| | 0 GeForce GTX 108... Off | 00000000:03:00.0 Off | N/A | | 29% 40C P0 52W / 250W | 0MiB / 11172MiB | 0% Default | +-------------------------------+----------------------+----------------------+ | 1 GeForce GTX 108... Off | 00000000:04:00.0 Off | N/A | | 30% 44C P0 48W / 250W | 0MiB / 11172MiB | 1% Default | +-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+ | Processes: GPU Memory | | GPU PID Type Process name Usage | |==============================================| | No running processes found | +-----------------------------------------------------------------------------+

sunzk2017 commented 6 years ago

problem has been solved after recompiling tensorflow with gpu option. thank a lot..