Closed xxxiaol closed 6 years ago
I simply use CUDA_VISIBLE_DEVICES to restrict it on GPU, but it does not work. I'm new to GPU, could you please tell me how to run the code on GPU? Thanks!
try installing tensorflow-gpu
I simply use CUDA_VISIBLE_DEVICES to restrict it on GPU, but it does not work. I'm new to GPU, could you please tell me how to run the code on GPU? Thanks!