Skip to content

Conversation

@weiji14
Copy link
Member

@weiji14 weiji14 commented Apr 1, 2025

Consolidate three conda specification files into one environment.yml file. Notable changes are:

See also #28, part of #29

Consolidate two conda specification files into one environment.yml file. Notable changes are:
- Staying on Python 3.11 instead of Python 3.12
- Drop torchvision/torchaudio as not used in this repo
- Using `nvidia-dali-python` from conda-forge instead of `nvidia-dali-cuda120` from PyPI
- Using nightly `kvikIO=25.04.00a` instead of https://github.com/TomAugspurger/kvikio-zarr-v3
- Using `xarray>=2025.3.0` containing pydata/xarray#10078
@weiji14 weiji14 self-assigned this Apr 1, 2025
@weiji14 weiji14 changed the title Merge zarr-gpu-env.yaml into environment.yml Merge conda environment.yml files Apr 1, 2025
@weiji14 weiji14 marked this pull request as ready for review April 1, 2025 23:02
@weiji14 weiji14 merged commit 3a8699a into main Apr 1, 2025
@weiji14 weiji14 deleted the merge-conda-envs branch April 1, 2025 23:12
@weiji14 weiji14 mentioned this pull request Apr 1, 2025
8 tasks
Copy link

@jakirkham jakirkham left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks Wei! 🙏

Had a few questions/comments below

dependencies:
- cupy~=13.3.0
- dask-jobqueue~=0.9.0
- rapidsai-nightly::kvikio>=25.04.00a

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes, I'm keeping a close eye on that release, and will update this once it's out!

- jupyterlab~=4.3.5
- nvidia-dali-python~=1.45.0
- python=3.11
- nvtx~=0.2.11

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Am curious about the context for the nvtx version constraint

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Probably ok to unpin, I just applied a ~= constraint for consistency in this environment.yml file

- xarray>=2025.3.0
- zarr~=3.0.3
- pip:
- nvidia-nvcomp-cu12

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

FWIW we have nvcomp packages on conda-forge

Though those are C++ only atm. If you need the Python bits, please let us know with a new issue on that feedstock

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yep, that would be nice! Asked over at conda-forge/nvcomp-feedstock#21

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants