Skip to content
This repository has been archived by the owner on Aug 3, 2021. It is now read-only.

GPU is not being used? #562

Open
Ramnathan-zoimeet opened this issue May 26, 2021 · 0 comments
Open

GPU is not being used? #562

Ramnathan-zoimeet opened this issue May 26, 2021 · 0 comments

Comments

@Ramnathan-zoimeet
Copy link

I did follow the instruction as per requirement i installed CUDA 10.2, and TensorFlow - 1.15.4 and Python 3.6.9 and to make sure all installation went right i did run the run_all_tests.sh and it was successful with no errors, but still when i run the main script with train/ infer/ eval any of these modes the GPU is being assigned right but it does not use it or speed up the process (using a single 16gb v100 GPU and tried with horovod ), so is there anything else i need to do to make the GPU work?

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant