异构计算实战2：WSL2子系统Ubuntu安装CUDA

原创已于 2025-05-17 09:06:35 修改 · 1.4k 阅读

20 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

收录于

异构计算实战

于 2025-05-17 08:56:37 首次发布

AudioSeal 音频水印系统

**AudioSeal** 是 Meta 开源的语音水印系统，用于 AI 生成音频的检测和溯源。

本文是《异构计算实战》系列文章第2篇。

环境准备：WSL2子系统Ubuntu安装CUDA

继 Windows 11安装Linux子系统WSL2 之后，接下来我们为Ubuntu 22.04配置CUDA编程环境。

一、Windows安装NVIDIA显卡驱动

这一步很重要，一定要确保Windows正确安装显卡驱动，否则后续步骤总是会出现奇奇怪怪的问题。

Windows安装NVIDIA显卡驱动，以GeForce MX350显卡为例。

访问NVIDIA官网，搜索适合的显卡驱动：

中文：https://www.nvidia.cn/geforce/drivers/

英文：https://www.nvidia.com/Download/index.aspx

这里下载最新版驱动：556.12

安装完成后，在Windows command中查询CUDA信息：

C:\Users\yz> nvidia-smi
Wed Jul 10 21:25:48 2024
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 556.12                 Driver Version: 556.12         CUDA Version: 12.5     |
|-----------------------------------------+------------------------+----------------------+
| GPU  Name                  Driver-Model | Bus-Id          Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |           Memory-Usage | GPU-Util  Compute M. |
|                                         |                        |               MIG M. |
|=========================================+========================+======================|
|   0  NVIDIA GeForce MX350         WDDM  |   00000000:01:00.0 Off |                  N/A |
| N/A   51C    P0             N/A / ERR!  |       0MiB /   2048MiB |      2%      Default |
|                                         |                        |                  N/A |
+-----------------------------------------+------------------------+----------------------+

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

二、Ubuntu安装CUDA Toolkit

注意，这里我们安装的是“CUDA Toolkit”，WSL使用的CUDA驱动实际上是Windows的，所以子系统的Ubuntu不需安装显卡驱动。

访问NVIDIA官网，选择合适的安装方式。

CUDA Toolkit官网

我们选择：CUDA Toolkit 12.5.0

然后，选择“Operating System”，“Architecture”，“Distribution”，“Version”，“Installer Type”。

本文推荐使用“Installer Type”使用“runfile (local)”方式。

选择安装方式后，按照页面给出的步骤执行安装步骤。

执行.run程序之前需要先安装gcc：

apt install gcc
apt install gcc++

安装gcc后，执行上述.run程序，然后在以下步骤输入“accept”并回车继续。

接下来选择需要安装的组件。这里不用选择“Kernel Object”。然后选择“Install”，并回车继续安装。

安装成功，可以看到以下提示信息：

安装成功后，按照提示在用户的.bashrc文件中添加环境变量：

export CUDA_HOME=/usr/local/cuda-12.5
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=$CUDA_HOME/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

检查CUDA是否正确安装：

root@FF:~/cuda# nvidia-smi
Wed Jul 10 21:49:44 2024
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 555.58.02              Driver Version: 556.12         CUDA Version: 12.5     |
|-----------------------------------------+------------------------+----------------------+
| 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. |
|=========================================+========================+======================|
segmentation fault

root@FF:~/cuda# nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Wed_Apr_17_19:19:55_PDT_2024
Cuda compilation tools, release 12.5, V12.5.40
Build cuda_12.5.r12.5/compiler.34177558_0