search

JiahuiYu / neuralgym

Deep Learning Toolkit
MIT License
101 stars 97 forks source link

problem about np.get_gpus() #4

Closed Irvingwangjr closed 6 years ago

Irvingwangjr commented 6 years ago

hi jiahui,i'am using your inpainting test model. when the code comes to this np.get_gpus(1) line, an weird error happen. 42c09b862b5bacffad36b09f79e1b69

i was wandering whether my environment was wrong: i'm using anaconda2 with py3.5 +cuda9.0+cudnn7.1 my GPU is 1050Ti.

JiahuiYu commented 6 years ago

Just remove that line. The reason is because Xorg and other programs are taken up your GPU. ng.get_gpus(1) will automatically get one FREE gpu.