Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Apple Silicon GPU compatibility for Tensorflow #2184

Merged
merged 3 commits into from
Oct 21, 2024
Merged

Apple Silicon GPU compatibility for Tensorflow #2184

Changes from 2 commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
597f490
added dependencies for installation of tf for m1-m2 mac
comane Oct 20, 2024
ceda0c8
added missing doc on savepseudodata
comane Oct 20, 2024
d745186
added link to issue
comane Oct 21, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 12 additions & 3 deletions doc/sphinx/source/n3fit/runcard_detailed.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -332,9 +332,9 @@ top-level option:

parallel_models: true


And then run ``n3fit`` with a replica range to be parallelized
(in this case from replica 1 to replica 4).
Note that currently, in order to run with parallel models, one has to set ``savepseudodata: false``
in the ``fitting`` section of the runcard. Once this is done, the user can run ``n3fit`` with a
replica range to be parallelized (in this case from replica 1 to replica 4).

.. code-block:: bash

Expand All @@ -346,6 +346,15 @@ should run by setting the environment variable ``CUDA_VISIBLE_DEVICES``
to the right index (usually ``0, 1, 2``) or leaving it explicitly empty
to avoid running on GPU: ``export CUDA_VISIBLE_DEVICES=""``

Note that in order to run the replicas in parallel using the GPUs of a M1/M2 Mac, it is necessary to also install
comane marked this conversation as resolved.
Show resolved Hide resolved
the following packages:

.. code-block:: bash

conda install -c apple tensorflow-deps
pip install tensorflow-macos==2.13.0 tensorflow-metal wandb==0.15.9
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why is it necessary to really pin this version?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It is related to this issue here: wandb/wandb#5935

I was not able to make it run on MaC M2 GPUs with other versions.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Could you also add a reference to that issue in the docs?




.. _otheroptions-label:

Expand Down