PyNET-Bokeh icon indicating copy to clipboard operation
PyNET-Bokeh copied to clipboard

Published 20 hours ago • verified publisher icon aiff22

Unable to load gpu for train？

Open wuguowuge opened this issue 5 years ago • 2 comments

I use ：os.environ["CUDA_VISIBLE_DEVICES"] = " "，but it was not work？

wuguowuge avatar Jul 01 '20 11:07 wuguowuge

Hi @wuguowuge,

The provided codes are running on GPU by default. Please make sure that you have CUDA and cuDNN drivers installed in your system, and that you installed the correct TensorFlow-GPU version (pip install tensorflow-gpu).

aiff22 avatar Jul 15 '20 10:07 aiff22

I have encountered the same problem and cannot call the GPU. How can I solve it?

C-water avatar Apr 05 '21 03:04 C-water