Description
NumPy 2.0 is coming out soon ( numpy/numpy#24300 ). NumPy 2.0.0rc1 packages for conda & wheels came out 2 weeks back ( numpy/numpy#24300 (comment) )
Ecosystem support for NumPy 2.0 is being tracked in issue: numpy/numpy#26191
Also conda-forge is discussing how to support NumPy 2.0: conda-forge/conda-forge.github.io#1997
When building against NumPy 2.0, it is possible with default settings to build packages that are compatible with NumPy 1 & 2. Where NumPy will target the oldest NumPy version that was built for that Python version being targeted
Developed the following list by installing RAPIDS 24.04 and inspecting, which packages used NumPy. Specifically ran the commands below
conda install -n base conda-tree -y
conda create -n rapids-24.04 -c rapidsai -c conda-forge -c nvidia rapids=24.04 python=3.11 cuda-version=12.2 -y
conda tree -n rapids-24.04 whoneeds numpy
This generated a list of dependencies. Some of these were RAPIDS packages themselves. So removed those from the list. Also dropped some indirect dependencies of RAPIDS. Admittedly this can get a little subjective. Though tried to capture a sufficiently complete, though not overly detailed, picture
From this, have built the table below
Some versions have questions marks if...
- The release is uncertain
- There may be some lingering issues
- It may have been fixed against an older NumPy 2, but hasn't been retested with RC1
Blank entries mean no information is known about those fields at this time
Note to editors: Also attaching the CSV file used to generate this table (as editing Markdown tables can be tricky 😅). Would suggest making any changes in the CSV file locally (with Excel or other). Then you can use prettytable
(available on PyPI & Conda-forge) to generate Markdown with this code. The resulting content can be copy-pasted above. Can drag and drop the CSV file into this textbox to attach it
prettytable
code:
import prettytable
with open("rapids_numpy_pkgs.csv", "r") as fh:
t = prettytable.from_csv(fh, delimeter=",", lineterminator="\n")
t.set_style(prettytable.MARKDOWN)
with open("rapids_numpy_pkgs.md", "w") as fh2:
fh2.write(str(t))
Activity