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

support cuda toolkit version: 11.7 #214

Merged

Conversation

wenhaozhao
Copy link
Contributor

This PR adds support for CUDA Toolkit 11.7

@coreylowman
Copy link
Owner

This looks great! Just will need to add nccl & cudnn.

You can extract these inside docker pull nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04 if you don't have access to nccl/cudnn on whatever machine you produced these on!

add cudnn & nccl
1 cudnn/sys/sys_11070.rs
2 nccl/sys/sys_11070.rs
@wenhaozhao wenhaozhao force-pushed the feature-support-cuda_toolkit_11.7 branch from 668be1f to 241d378 Compare March 28, 2024 02:42
@wenhaozhao
Copy link
Contributor Author

This looks great! Just will need to add nccl & cudnn.

You can extract these inside docker pull nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04 if you don't have access to nccl/cudnn on whatever machine you produced these on!

ok, thanks!
I've added two files about nccl & cudnn

  • 1 src/cudnn/sys/sys_11070.rs
  • 2 src/nccl/sys/sys_11070.rs

@coreylowman
Copy link
Owner

Yay perfect, thanks

@coreylowman coreylowman merged commit c45c1d2 into coreylowman:main Mar 28, 2024
3 checks passed
@coreylowman
Copy link
Owner

Hmm looks like there were some pretty breaking changes for cudnn/nccl on 11.7 (getting errors when trying to compile for that version in the safe apis). Will need to think about that

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.

None yet

2 participants