requirements_gpu.txt missing / deprecated?
joedeveloper opened this issue · comments
Thank you so much for this!
This is the first tutorial re ML I have undertaken and it was an absolute pleasure.
The model training time was expected, but I'd like to do what I can to reduce it going forwards.
In the readme.md it mentions a requirements_gpu.txt file, any hints?
Hi Joe,
Thank you for your interest! I would happy to help you.
You mentioned the requirements_gpu.txt file that I forgot to commit. It is done now ;)
If you want to train the model using an Nvidia GPU you need to install CUDA drivers and tensorflow-gpu library. Please have a look at this link as it may help you configure your environement https://www.tensorflow.org/install/gpu
Best,