Skip to content

Resolves bugs and better supports Cuda libraries like PyTorch and DeepSpeed #1360

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

Merged
merged 1 commit into from
Feb 1, 2024

Conversation

lh0x00
Copy link
Contributor

@lh0x00 lh0x00 commented Feb 1, 2024

Adding Cuda home environment values helps other libraries avoid unwanted errors, for examples PyTorch and DeepSpeed.
PyTorch and DeepSpeed is just libraries I want to use as examples, ref:

Please review my PR, thanks for all, @rosbo @djherbis

Copy link

google-cla bot commented Feb 1, 2024

Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this failed invocation of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.

@lh0x00 lh0x00 changed the title Resolves bugs and better supports Cuda libraries like DeepSpeed Resolves bugs and better supports Cuda libraries like PyTorch and DeepSpeed Feb 1, 2024
@rosbo
Copy link
Contributor

rosbo commented Feb 1, 2024

We indeed currently do not set this path and I also confirmed the path points to the correct CUDA path:
BSXufTSXXpMFY2g

Thank you

@rosbo rosbo merged commit 71487b6 into Kaggle:main Feb 1, 2024
@rosbo rosbo self-assigned this Feb 1, 2024
@rosbo rosbo self-requested a review February 1, 2024 16:43
@lh0x00
Copy link
Contributor Author

lh0x00 commented Feb 1, 2024

@rosbo thanks for reviewed

@lh0x00 lh0x00 deleted the bugfix/DeepSpeedIssues branch February 2, 2024 02:53
calderjo pushed a commit that referenced this pull request Feb 6, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants