dreyeve icon indicating copy to clipboard operation
dreyeve copied to clipboard

how to use GPU? which version of Python, cuda, keras, and Theano should I use

Open HHUBRO opened this issue 11 months ago • 0 comments

I want to inference with the pre-trained model, which version of Python, cuda, keras, and Theano should I use. Besides, Do you use the system-level cuda or the cuda in the virtual environment? I have tried with python=2.7, keras=1.1.0,cuda=10.0,theano=1.0.4,if I inference with GPU, the result obviouly wrong(show 阿as fig), while correct with the cpu,how can I fix it? image

HHUBRO avatar Mar 24 '24 14:03 HHUBRO