search

tryolabs / luminoth

Deep Learning toolkit for Computer Vision.
https://tryolabs.com
BSD 3-Clause "New" or "Revised" License
2.4k stars 399 forks source link

Predict not working on GPU #209

Open EthanReid opened 6 years ago

EthanReid commented 6 years ago

Hi,

I got the predict function to work with only the cpu, but when I run with gpu it just returns nothing. This is what I am getting in the terminal. C:\Users\Evan\Desktop> lumi predict vids/lunch.mp4 --checkpoint=fast --save-media-to=vidO Found 1 files to predict. 2018-08-17 19:19:41.978750: I T:\src\github\tensorflow\tensorflow\core\platform\cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 2018-08-17 19:19:42.234390: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1356] Found device 0 with properties: name: TITAN X (Pascal) major: 6 minor: 1 memoryClockRate(GHz): 1.531 pciBusID: 0000:01:00.0 totalMemory: 12.00GiB freeMemory: 9.93GiB 2018-08-17 19:19:42.311618: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1356] Found device 1 with properties: name: GeForce GTX 1070 major: 6 minor: 1 memoryClockRate(GHz): 1.683 pciBusID: 0000:02:00.0 totalMemory: 8.00GiB freeMemory: 6.63GiB 2018-08-17 19:19:42.315542: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1435] Adding visible gpu devices: 0, 1 2018-08-17 19:19:43.325791: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:923] Device interconnect StreamExecutor with strength 1 edge matrix: 2018-08-17 19:19:43.328053: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:929] 0 1 2018-08-17 19:19:43.329931: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:942] 0: N N 2018-08-17 19:19:43.331604: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:942] 1: N N 2018-08-17 19:19:43.333030: I T:\src\github\tensorflow\tensorflow\core\common_runtime\gpu\gpu_device.cc:1053] Created TensorFlow device (/job:localhost/replica:0/task:0/device:GPU:0 with 9616 MB memory) -> physical GPU (device: 0, name: TITAN X (Pascal), pci bus id: 0000:01:00.0, compute capability: 6.1)

nagitsu commented 6 years ago

Does it hang there, does it return and error or just returns and nothing else? Does the vid0 directory exist?

dekked commented 5 years ago

@EthanReid any updates on this? Could you get it to work?