Problems encountered when python3 run.py

[kevin1121]

How do I solve these two problems?

Thanks for everyone's help!

[Aareon]

Did you pip install -r requirements.txt from within a venv?

[kevin1121]

Did you pip install -r requirements.txt from within a venv?

Thank you for your reply. I have executed the installation command according to the documentation. Two warnings were reported during the installation. Does this have any impact? How can I solve it?

[harindersinghk]

what version of pip are you using?

[kevin1121]

what version of pip are you using?

pip version is 24.0

[harindersinghk]

I assume you are using mac as wheels are primarily built for Linux. There seems to be compatibility issue with macOs. Which cuda have you installed?

Run nvidia-smi

[kevin1121]

I assume you are using mac as wheels are primarily built for Linux. There seems to be compatibility issue with macOs. Which cuda have you installed?

Run nvidia-smi

I am using a mac M1 chip, how should I run it?

[cassius425]

on M2 max , python3

[Aareon]

Theres a config option to change the mesh size. Also, consider installing one of the backends.

[Aareon]

From #292

See

https://github.com/xai-org/grok-1/blob/7050ed204b8206bb8645c7b7bbef7252f79561b0/run.py#L60

and change local_mesh_config=(1, 1)

[Luffy-Yao]

Try the following command pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html