在 WSL 上启用 NVIDIA CUDA

在 WSL 上启用 NVIDIA CUDA_第1张图片

环境要求

  1. Windows 11 或 Windows 10 版本 21H2
  2. 特定版本的GPU驱动:
    安装支持 NVIDIA CUDA 的 WSL 驱动程序:
https://www.nvidia.com/download/index.aspx

具体安装哪个版本,查阅:https://docs.nvidia.com/cuda/wsl-user-guide/index.html#getting-started-with-cuda-on-wsl

https://developer.nvidia.com/cuda/wsl

  1. WSL
    通过在“设置”应用的“Windows 更新”部分中选择“检查更新”,确保您拥有最新的内核。
    需要 5.10.43.3 或更高版本的内核版本。可以通过在 PowerShell 中运行以下命令来检查版本号。
wsl cat /proc/version

资源

https://docs.nvidia.com/cuda/wsl-user-guide/index.html#getting-started-with-cuda-on-wsl

Latest Linux CUDA toolkit package - WSL-Ubuntu from 12.x releases can be downloaded from
https://developer.nvidia.com/cuda-downloads

https://learn.microsoft.com/en-us/windows/ai/directml/gpu-cuda-in-wsl

debug:
https://forums.developer.nvidia.com/c/accelerated-computing/cuda/cuda-on-windows-subsystem-for-linux-wsl-2/303

注意:

nvidia-docker不再支持包装器,并且 NVIDIA Container Toolkit 已进行扩展,允许用户配置 Docker 以使用 NVIDIA Container Runtime 。

https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/index.html

你可能感兴趣的:(深度学习)