AI-Hypercomputer / AI-Hypercomputer/RecML
Jax CPU version is installed with the requirements.txt
Đang mở
- Ngôn ngữ chính
- Python
- Star
- 226
- Fork
- 22
- Merge trung bình
- 15 giờ 19 phút
- Pull request đã merge (30 ngày)
- 2
Mô tả
WARNING:2025-04-06 19:31:14,948:jax._src.xla_bridge:971: A Google TPU may be present on this machine, but either a TPU-enabled jaxlib or libtpu is not installed. Falling back to cpu.
W0406 19:31:14.948704 125696970082304 xla_bridge.py:971] A Google TPU may be present on this machine, but either a TPU-enabled jaxlib or libtpu is not installed. Falling back to cpu.
The above can be fixed by unintsalling the jax that comes with flax and installing the jax[tpu].
pip3 install jax[tpu]
Hướng dẫn đóng góp
Đánh giá
Issue này chưa được đánh giá.