How to run on GPU

Question

harshitazilen opened this issue 5 years ago · comments

I tried configuring below parameters but it still does not use GPU. Please guide:

os.environ["CUDA_DEVICE_ORDER"] = "PCI:1:0:0"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

Thanks

Rodrigo Meira de Andrade · Answer 1 · Mon May 06 2019 08:07:27 GMT+0800 (China Standard Time)

if you installed tensorflow correctly, it will use GPU by default

look4pritam · Answer 2 · Mon May 06 2019 21:51:09 GMT+0800 (China Standard Time)

Install cuda 10
Install cudnn for cuda-10
Install tensorflow- GPU.
Then it will work.

harshitazilen · Answer 3 · Tue May 07 2019 13:14:33 GMT+0800 (China Standard Time)

Thanks. I tried by creating new virtual environment and re-configured everything and it is working now.