From f530fd456bdf072659684cf900572f9b8ef44182 Mon Sep 17 00:00:00 2001 From: Jim Crist-Harif Date: Wed, 15 Jan 2025 20:45:02 +0000 Subject: [PATCH] Add aarch64 conda environments Previously we only had environment files for x86-64, which made setting up a dev environment on an aarch64 machine trickier. --- .../all_cuda-118_arch-aarch64.yaml | 82 +++++++++++++++++++ .../all_cuda-125_arch-aarch64.yaml | 78 ++++++++++++++++++ dependencies.yaml | 2 +- 3 files changed, 161 insertions(+), 1 deletion(-) create mode 100644 conda/environments/all_cuda-118_arch-aarch64.yaml create mode 100644 conda/environments/all_cuda-125_arch-aarch64.yaml diff --git a/conda/environments/all_cuda-118_arch-aarch64.yaml b/conda/environments/all_cuda-118_arch-aarch64.yaml new file mode 100644 index 0000000000..e543b2d0a5 --- /dev/null +++ b/conda/environments/all_cuda-118_arch-aarch64.yaml @@ -0,0 +1,82 @@ +# This file is generated by `rapids-dependency-file-generator`. +# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. +channels: +- rapidsai +- rapidsai-nightly +- dask/label/dev +- conda-forge +- nvidia +dependencies: +- c-compiler +- certifi +- cmake>=3.26.4,!=3.30.0 +- cuda-python>=11.8.5,<12.0a0 +- cuda-version=11.8 +- cudatoolkit +- cudf==25.2.*,>=0.0.0a0 +- cupy>=12.0.0 +- cuvs==25.2.*,>=0.0.0a0 +- cxx-compiler +- cython>=3.0.0 +- dask-cuda==25.2.*,>=0.0.0a0 +- dask-cudf==25.2.*,>=0.0.0a0 +- dask-ml +- doxygen=1.9.1 +- fmt>=11.0.2,<12 +- gcc_linux-64=11.* +- graphviz +- hdbscan>=0.8.39,<0.8.40 +- hypothesis>=6.0,<7 +- ipykernel +- ipython +- joblib>=0.11 +- libcublas-dev=11.11.3.6 +- libcublas=11.11.3.6 +- libcufft-dev=10.9.0.58 +- libcufft=10.9.0.58 +- libcumlprims==25.2.*,>=0.0.0a0 +- libcurand-dev=10.3.0.86 +- libcurand=10.3.0.86 +- libcusolver-dev=11.4.1.48 +- libcusolver=11.4.1.48 +- libcusparse-dev=11.7.5.86 +- libcusparse=11.7.5.86 +- libcuvs==25.2.*,>=0.0.0a0 +- libraft-headers==25.2.*,>=0.0.0a0 +- librmm==25.2.*,>=0.0.0a0 +- nbsphinx +- ninja +- nltk +- numba>=0.57 +- numpy>=1.23,<3.0a0 +- numpydoc +- nvcc_linux-aarch64=11.8 +- packaging +- pydata-sphinx-theme!=0.14.2 +- pylibraft==25.2.*,>=0.0.0a0 +- pynndescent +- pytest-benchmark +- pytest-cases +- pytest-cov +- pytest-xdist +- pytest==7.* +- python>=3.10,<3.13 +- raft-dask==25.2.*,>=0.0.0a0 +- rapids-build-backend>=0.3.0,<0.4.0.dev0 +- rapids-dask-dependency==25.2.*,>=0.0.0a0 +- recommonmark +- rmm==25.2.*,>=0.0.0a0 +- scikit-build-core>=0.10.0 +- scikit-learn==1.5.* +- scipy>=1.8.0 +- seaborn +- spdlog>=1.14.1,<1.15 +- sphinx +- sphinx-copybutton +- sphinx-markdown-tables +- statsmodels +- sysroot_linux-aarch64==2.28 +- treelite==4.3.0 +- umap-learn==0.5.6 +- xgboost>=2.1.0 +name: all_cuda-118_arch-aarch64 diff --git a/conda/environments/all_cuda-125_arch-aarch64.yaml b/conda/environments/all_cuda-125_arch-aarch64.yaml new file mode 100644 index 0000000000..7a64ef0c28 --- /dev/null +++ b/conda/environments/all_cuda-125_arch-aarch64.yaml @@ -0,0 +1,78 @@ +# This file is generated by `rapids-dependency-file-generator`. +# To make changes, edit ../../dependencies.yaml and run `rapids-dependency-file-generator`. +channels: +- rapidsai +- rapidsai-nightly +- dask/label/dev +- conda-forge +- nvidia +dependencies: +- c-compiler +- certifi +- cmake>=3.26.4,!=3.30.0 +- cuda-cudart-dev +- cuda-nvcc +- cuda-profiler-api +- cuda-python>=12.6.2,<13.0a0 +- cuda-version=12.5 +- cudf==25.2.*,>=0.0.0a0 +- cupy>=12.0.0 +- cuvs==25.2.*,>=0.0.0a0 +- cxx-compiler +- cython>=3.0.0 +- dask-cuda==25.2.*,>=0.0.0a0 +- dask-cudf==25.2.*,>=0.0.0a0 +- dask-ml +- doxygen=1.9.1 +- fmt>=11.0.2,<12 +- gcc_linux-aarch64=13.* +- graphviz +- hdbscan>=0.8.39,<0.8.40 +- hypothesis>=6.0,<7 +- ipykernel +- ipython +- joblib>=0.11 +- libcublas-dev +- libcufft-dev +- libcumlprims==25.2.*,>=0.0.0a0 +- libcurand-dev +- libcusolver-dev +- libcusparse-dev +- libcuvs==25.2.*,>=0.0.0a0 +- libraft-headers==25.2.*,>=0.0.0a0 +- librmm==25.2.*,>=0.0.0a0 +- nbsphinx +- ninja +- nltk +- numba>=0.57 +- numpy>=1.23,<3.0a0 +- numpydoc +- packaging +- pydata-sphinx-theme!=0.14.2 +- pylibraft==25.2.*,>=0.0.0a0 +- pynndescent +- pytest-benchmark +- pytest-cases +- pytest-cov +- pytest-xdist +- pytest==7.* +- python>=3.10,<3.13 +- raft-dask==25.2.*,>=0.0.0a0 +- rapids-build-backend>=0.3.0,<0.4.0.dev0 +- rapids-dask-dependency==25.2.*,>=0.0.0a0 +- recommonmark +- rmm==25.2.*,>=0.0.0a0 +- scikit-build-core>=0.10.0 +- scikit-learn==1.5.* +- scipy>=1.8.0 +- seaborn +- spdlog>=1.14.1,<1.15 +- sphinx +- sphinx-copybutton +- sphinx-markdown-tables +- statsmodels +- sysroot_linux-aarch64==2.28 +- treelite==4.3.0 +- umap-learn==0.5.6 +- xgboost>=2.1.0 +name: all_cuda-125_arch-aarch64 diff --git a/dependencies.yaml b/dependencies.yaml index 6761744857..6aa471a1bb 100644 --- a/dependencies.yaml +++ b/dependencies.yaml @@ -4,7 +4,7 @@ files: output: conda matrix: cuda: ["11.8", "12.5"] - arch: [x86_64] + arch: [x86_64, aarch64] includes: - common_build - cuda