PyPi torch+gpu fix - Githubissues

tumaer / lagrangebench

LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite

https://lagrangebench.readthedocs.io

MIT License

57 stars 6 forks source link

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.