Enable GPU for local Jupyter notebook

[david-ben-gurion]

Hi,
Is there any way to enable on-GPU computations when working with local jupyter notebooks? Right now, the epochs run slow on large data and using commands I have found out that the CPU is being used instead of the GPU. Yes, I have a compatible card.

[SadanandaJaya]

Hi David,

I am not an expert in here but found this resource thought this might help you.

https://stackoverflow.com/questions/51002045/how-to-make-jupyter-notebook-to-run-on-gpu

[mrdbourke]

Hey @DavidBenGurion-Git,

You can see if your GPU is in use using the guide here: https://www.tensorflow.org/guide/gpu

If you need to setup TensorFlow for GPU locally, see here: https://www.tensorflow.org/install/gpu

This comment on GitHub also helps setup a TensorFlow environment using Conda: tensorflow/tensorflow#20271 (comment)