在 WSL2 和 Hyper-V 中使用 Nvidia Cuda 功能

在 Windows 上直接测试应用一些 AI 程序，有时候会遇到一些不兼容问题，如果是生产环境，直接用 Linux 系统就解决问题。 如果需要在测试或者验证过程中在 Windows 下可以使用 WSL2 和 Hyper-V 安装 Linux 系统来做替代。

注意事项

  • Linux 版本推荐使用 Ubuntu 22.04 LTS , 因为 nvidia WSL cuda 目前只提供了 ubuntu 的包。
  • 目前 nvidai 官方表示只支持 geforce 显卡。

WSL2 支持 cuda

  • Nvidia 官方说明

https://docs.nvidia.com/cuda/wsl-user-guide/index.html

  • CUDA 下载地址

https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=WSL-Ubuntu&target_version=2.0&target_type=deb_local

Hyper-V 支持 cuda

Hyper-V 的支持主要有一下两个 github 仓库说明:

  1. https://github.com/brokeDude2901/dxgkrnl_ubuntu/
  2. https://github.com/5l1v3r1/dxgkrnl_ubuntu

第二个其实是 fork 第一个，做了一些其他说明。