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

Add experimental support for CUDA under WSL for Windows 11 #2538

Merged
merged 2 commits into from
Oct 23, 2021

Conversation

wjsi
Copy link
Member

@wjsi wjsi commented Oct 21, 2021

What do these changes do?

Add support for CUDA under WSL2 for Windows 11. As NVML library not fully supported under WSL2, we use XML results from nvidia-smi command to obtain device details instead.

Related issue number

Fixes #2537

@wjsi wjsi added type: feature New feature to be backported Indicate that the PR need to be backported to stable branch hw: nvidia gpu labels Oct 21, 2021
@wjsi wjsi added this to the v0.8.0rc1 milestone Oct 21, 2021
@wjsi wjsi requested a review from hekaisheng as a code owner October 21, 2021 14:55
@wjsi wjsi requested a review from qinxuye as a code owner October 21, 2021 14:55
@wjsi wjsi changed the title Add support for CUDA under WSL for Windows 11 Add experimental support for CUDA under WSL for Windows 11 Oct 21, 2021
Copy link
Collaborator

@qinxuye qinxuye left a comment

Choose a reason for hiding this comment

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

LGTM

Copy link
Contributor

@hekaisheng hekaisheng left a comment

Choose a reason for hiding this comment

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

LGTM

@hekaisheng hekaisheng merged commit a50689c into mars-project:master Oct 23, 2021
@wjsi wjsi deleted the enh/wsl_cuda branch October 23, 2021 05:02
@qinxuye qinxuye added backported already PR has been backported and removed to be backported Indicate that the PR need to be backported to stable branch labels Oct 23, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Enable GPU usage in WSL2 in Windows 11
3 participants