Skip to content

Commit

Permalink
Add aarch64 conda environments
Browse files Browse the repository at this point in the history
Previously we only had environment files for x86-64, which made setting
up a dev environment on an aarch64 machine trickier.
  • Loading branch information
jcrist committed Jan 15, 2025
1 parent cf259f6 commit c48816d
Show file tree
Hide file tree
Showing 3 changed files with 161 additions and 1 deletion.
82 changes: 82 additions & 0 deletions conda/environments/all_cuda-118_arch-aarch64.yaml
Original file line number Diff line number Diff line change
@@ -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-aarch64=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.17
- treelite==4.3.0
- umap-learn==0.5.6
- xgboost>=2.1.0
name: all_cuda-118_arch-aarch64
78 changes: 78 additions & 0 deletions conda/environments/all_cuda-125_arch-aarch64.yaml
Original file line number Diff line number Diff line change
@@ -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=11.*
- 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.17
- treelite==4.3.0
- umap-learn==0.5.6
- xgboost>=2.1.0
name: all_cuda-125_arch-aarch64
2 changes: 1 addition & 1 deletion dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ files:
output: conda
matrix:
cuda: ["11.8", "12.5"]
arch: [x86_64]
arch: [x86_64, aarch64]
includes:
- common_build
- cuda
Expand Down

0 comments on commit c48816d

Please sign in to comment.