issues
search
tumaer
/
lagrangebench
LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite
https://lagrangebench.readthedocs.io
MIT License
57
stars
6
forks
source link
PyPi torch+gpu fix
#9
Closed
gerkone
closed
1 year ago
gerkone
commented
1 year ago
Workaround for the PyPi/Wheel issue that installed the entire torch+CUDA runtime when only torch CPU is needed.
Explicit install instructions
Set JAX version to
0.4.14
0.4.18
(before
jax.abstract_arrays
refactor).
Needed by JAX-MD.
arturtoshev
commented
1 year ago
LGTM.
0.4.14
0.4.18
(beforejax.abstract_arrays
refactor).Needed by JAX-MD.