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

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

win主机安装CUDA Toolkit:

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                                                 |
+-----------------------------------------------------------------------------+

参考:

纠错,疑问,交流: 请进入讨论区点击加入Q群

获取最新文章: 扫一扫右上角的二维码加入“创客智造”公众号


标签: windows系统入门教程