Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

update version of cupy to be 12.0.0 minimum #899

Merged
merged 1 commit into from
Apr 7, 2023

Conversation

jperez999
Copy link
Collaborator

This PR updates the version of cupy on the containers to make sure it is 12.0 or later. This is important because in v12.0+ dlpack is implemented correctly. This is leveraged by tensortable.

@jperez999 jperez999 added bug Something isn't working chore Infrastructure update ci labels Apr 7, 2023
@jperez999 jperez999 added this to the Merlin 23.04 milestone Apr 7, 2023
@jperez999 jperez999 requested a review from karlhigley April 7, 2023 01:04
@jperez999 jperez999 self-assigned this Apr 7, 2023
@github-actions
Copy link

github-actions bot commented Apr 7, 2023

Documentation preview

https://nvidia-merlin.github.io/Merlin/review/pr-899

Comment on lines -107 to +109
pip install cupy-cuda115 -f https://pip.cupy.dev/aarch64/ ; \
pip install cupy-cuda11x -f https://pip.cupy.dev/aarch64/ ; \
else \
pip install cupy-cuda117 ; \
pip install cupy-cuda11x ; \
Copy link
Contributor

Choose a reason for hiding this comment

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

Maybe I'm missing something but shouldn't these be cupy-cuda12x not cuda11x if we want the minimum version to be 12.0?

Copy link
Contributor

@karlhigley karlhigley Apr 7, 2023

Choose a reason for hiding this comment

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

It's confusing, because CUDA and CuPy are both currently at version 12. We need CuPy version 12.0.0 or greater, which is the latest version available for CUDA 11.

@karlhigley karlhigley merged commit b5b06dd into NVIDIA-Merlin:main Apr 7, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working chore Infrastructure update ci
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants