I tried to install dependencies with PDM for a project and found the inconvenience that TensorFlow does not detect the GPU of the M1 Pro. When I create a virtual environment with poetry I do not have the same problem.
Any hint on how to solve this inconvenience?
Post
Replies
Boosts
Views
Activity
I have been working with TensorFlow on my M1, however I have tried several times to deactivate the logs from TensorFlow and nothing seems to work.
Could someone provide a way to deactivate them?
Hello,
I have recently bought the M1 Pro and installed Tensorflow using the indications of Apple.
I noted that training a simple NN with GPU was really slow (~30 secs per epoch). However, after forcing Tensorflow to use CPU I saw an incredible improvement (~3 secs per epoch).
I cannot believe that the difference is that much. Verified with the System Monitor and I saw Tensorflow using 100% of the GPU.
Could anyone assist me or enlighten me?
Please check below the code used to train this model.
Hello,
I have read in this Apple's post that the option for Multi-GPU is currently not supported. I would like to know if this means that currently is working on only one GPU core, and if this means the performance for the training of the models will improve in future updates.
Thanks in advance.