Open
Description
I notice in the pyproject.toml, the CUDA version is pinned to 12.8. Is this necessary for running the package?
When I manually edit the file to point to the PyTorch CUDA 12.4 versions, I am unable to install the package.
Repro:
- Create uv venv
- Update all references to 128 with 124 in pyproject.toml
uv sync
× No solution found when resolving dependencies:
╰─▶ Because there is no version of torch==2.7.0 and nemo-rl depends on torch==2.7.0, we can conclude that nemo-rl's requirements are unsatisfiable.
And because your workspace requires nemo-rl[mcore], we can conclude that your workspace's requirements are unsatisfiable.