Files
Mujoco_WASM/mjx/cuda_requirements.txt
T
Saran Tunyasuvunakool e18567545b Update various requirements.txt files.
PiperOrigin-RevId: 698009096
Change-Id: Iaa2cdadaf31c16c580ce5c2060c8bf2b8377956e
2024-11-19 07:32:00 -08:00

16 lines
1.0 KiB
Plaintext

jax[cuda12_local]==0.4.34; python_version >= '3.10' \
--hash=sha256:b957ca1fc91f7343f91a186af9f19c7f342c946f95a8c11c7f1e5cdfe2e58d9e
jax[cuda12_local]==0.4.30; python_version == '3.9' \
--hash=sha256:289b30ae03b52f7f4baf6ef082a9f4e3e29c1080e22d13512c5ecf02d5f1a55b
jax-cuda12-plugin==0.4.34; python_version >= '3.10' \
--hash=sha256:d035ea72bd9b8a65a6ea621bca1affdd33127fa3a52e7bded7692670d360adab \
--hash=sha256:db988b7ba5063483a936ddbf162f04d1b4412e0d64340f11788c7bbc877e8a43 \
--hash=sha256:e23721d1654b311b47cd6b35768520284bd036f8c7e6b11600143258b4a0409a \
--hash=sha256:b2099a4407225122ff76f6dcdc8dbdae47e6f29343bdfd21460ad337dc34a209
jax-cuda12-plugin==0.4.30; python_version == '3.9' \
--hash=sha256:d8d196241b9253ecb1144a4409b5deacbb9771624f097b2bbf025da3c7d8f4f8
jax-cuda12-pjrt==0.4.34; python_version >= '3.10' \
--hash=sha256:0c7cc98f962cc7fc8e0a5ea6331b42a0cee516f202f1c3019f6aa5cd9530cca0
jax-cuda12-pjrt==0.4.30; python_version == '3.9' \
--hash=sha256:895d0198ad99638fcaf976c47592e2a543eef79ea15fabd24a402d055390c328