`numpy` version requirement fix

Lightning-AI / lightning-thunder

Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once; across one or thousands of GPUs.

Apache License 2.0

1.07k stars 60 forks source link

`numpy` version requirement fix #611

Closed nikitaved closed 1 week ago

nikitaved commented 1 week ago

Fix for numpy 2.0 version.

t-vi commented 1 week ago

The versioning is fine (I did add that when numpy 2 first acted up), we need to fix the GPU docker recipe to fix this.