Windows系统入门教程-WSL2增加GPU支持

Windows系统支持入门教程-WSL2增加GPU支持

说明：

• 介绍如何在WSL2中使用GPU
• 环境：win10 22H2 + Geforce RTX3060Ti + wsl2 + ubuntu20.04

版本检查：

• 1. GPU Driver 版本
  • 直接升级到最新版本的吧，默认一般460多，470多的版本太老旧了
  • 需要大于471.21版本
  • 下载GPU驱动，可以根据自己的显卡下载对应版本
  • 打开Navida控制版查看版本，升级完，这里是528.02版本

• 2. WSL2内核版本
  • 需要大于5.10.43.3版本
  • 查看WSL2内核版本等信息，powershell运行wsl --status
  • 关闭开启的系统，powershell运行wsl --shutdown
  • wsl升级内核，powershell运行wsl --update，升级完，内核版本是： 5.10.102.1
• 3. 操作系统版本
  • win10版本使用最新的21H2或22H2
  • 或使用win11版本
• 4. 确认Linux系统分发版本为WSL2
  • 查看Linux系统分发版本是WSL1或WSL2， powershell运行wsl -l -v
  • 设置Linux为WSL2，powershell运行wsl --set-default Ubuntu-20.04 2

win主机安装CUDA Toolkit：

• 需要主机支持CUDA和cuDnn的可安装
• 1.查看本机的CUDA驱动适配版本
  • 通过Nvidia控制板 -> 帮助 -> 系统信息 -> 组件 查看NVCUDA64.DLL一行确定cuda需要的版本，这里是cuda 12.0.133
• 2. 下载win10 的CUDA Toolkit 12.0版本
  • 进入下载地址，下载cuda对应win10的版本
  • 程序比较大，可以采用迅雷等下载
• 3. 下载win10对应版本的cudnn
  • 进入下载地址，下载cudnn对应win10的版本
  • 这里需要注册和登录之后才能下载，程序比较大，可以采用迅雷等下载
• 详细教程，可以参考

WSL上安装CUDA Toolkit：

• 需要WSL支持CUDA的安装
• 安装CUDA Toolkit命令

sudo apt-key del 7fa2af80
wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pin
sudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.0.0/local_installers/cuda-repo-wsl-ubuntu-12-0-local_12.0.0-1_amd64.deb
sudo dpkg -i cuda-repo-wsl-ubuntu-12-0-local_12.0.0-1_amd64.deb
sudo cp /var/cuda-repo-wsl-ubuntu-12-0-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

• 添加配置到bashrc

echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PAT' >> ~/.bashrc

• WSL下测试nvcc

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Mon_Oct_24_19:12:58_PDT_2022
Cuda compilation tools, release 12.0, V12.0.76
Build cuda_12.0.r12.0/compiler.31968024_0

• WSL下测试nvidia-smi

$ nvidia-smi
Sat Jan 21 10:40:36 2023
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 527.92.01    Driver Version: 528.02       CUDA Version: 12.0     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  On   | 00000000:01:00.0  On |                  N/A |
|  0%   37C    P8    20W / 200W |    846MiB /  8192MiB |     11%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

参考：

• 资料1