NVIDIA-NeMo / NVIDIA-NeMo/RL

Cannot install NeMo RL on Mac due to no available Triton version

Open
#1,005 3 comments 0 reactions 0 assignees View on GitHub
bug
Dominant language
Python
Stars
2k
Forks
561
Avg merge
4d 5h
Merged PRs (30d)
145

Description

**Describe the bug**

I'm trying to do local development on NeMo RL on my local Mac. I'm not trying to run anything meaningful, just trying to develop on local since it's faster. But I rely heavily on VSCode type checking, which requires the NeMo RL environment to be installed.

**Steps/Code to reproduce bug**

```bash
uv venv
source .venv/bin/activate
uv pip install -e .
```
Output
```
(NeMo-RL) bxyu@bxyu-mlt NeMo-RL % uv pip install -e .
× No solution found when resolving dependencies:
╰─▶ Because only the following versions of triton are available:
triton==2.0.0
triton==2.1.0
triton==2.2.0
triton==2.3.0
triton==2.3.1
triton==3.0.0
triton==3.1.0
triton==3.2.0
triton==3.3.0
triton==3.3.1
triton==3.4.0
and triton<=2.1.0 has no wheels with a matching Python ABI tag (e.g., `cp312`), we can conclude that triton<=2.1.0 cannot be used.
And because triton>=2.2.0 has no wheels with a matching platform tag (e.g., `macosx_15_0_arm64`) and nemo-rl==0.4.0rc0 depends on triton,
we can conclude that nemo-rl==0.4.0rc0 cannot be used.
And because only nemo-rl==0.4.0rc0 is available and you require nemo-rl, we can conclude that your requirements are unsatisfiable.

hint: You require CPython 3.12 (`cp312`), but we only found wheels for `triton` (v2.1.0) with the following Python ABI tags: `cp37m`,
`cp38`, `cp39`, `cp310`, `cp311`

hint: Wheels are available for `triton` (v3.4.0) on the following platforms: `manylinux_2_27_aarch64`, `manylinux_2_27_x86_64`,
`manylinux_2_28_aarch64`, `manylinux_2_28_x86_64`
```

**Expected behavior**

I am able to pip install NeMo RL successfully.

**Environment overview (please complete the following information)**

- Environment location: Bare-metal
- Method of install: pip install

**Environment details**

If NVIDIA docker image is used you don't need to specify these.
Otherwise, please provide:
- OS version - I am on Mac
- PyTorch version
- Python version 3.12

**Additional context**

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.