JiahuiYu / neuralgym

Deep Learning Toolkit

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

problem about np.get_gpus()

Irvingwangjr opened this issue · comments

commented

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.

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.