如何安装 WSL2 CUDA 驱动
时间: 2025-03-21 10:15:20 浏览: 37
### 如何在WSL2上安装CUDA驱动
#### 准备工作
为了能够在Windows Subsystem for Linux (WSL2) 上成功安装 CUDA 驱动程序,需要先确认系统满足基本条件。这包括确保已启用 WSL2 功能以及安装支持的 NVIDIA 显卡驱动版本[^1]。
#### 下载NVIDIA Windows显卡驱动
访问官方链接 [NVIDIA Driver Downloads](https://www.nvidia.com/Download/index.aspx),找到适合您硬件配置的最新版驱动并完成安装。这是实现 GPU 加速计算的基础步骤之一。
#### 获取适用于WSL2的CUDA工具包
前往[NVIDIA Developer Cuda For WSL Page](https://developer.nvidia.com/cuda/wsl),依据个人设备上的具体显卡型号挑选对应的 CUDA 工具包进行下载与后续操作前准备活动[^3]。
#### 更新Ubuntu至最新状态
打开终端执行如下命令以保证软件仓库处于最前沿水平:
```bash
sudo apt update && sudo apt upgrade -y
```
#### 安装必要的依赖项
通过运行下面这段脚本来获取构建和编译过程中可能需要用到的各种库文件和支持组件:
```bash
sudo apt install build-essential cmake git libssl-dev pkg-config python-is-python3 python3-pip unzip wget -y
```
#### 设置环境变量
编辑 `~/.bashrc` 文件加入以下两行内容以便于之后能够顺利调用相关路径下的可执行文件:
```bash
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
```
随后重新加载该配置文档使其生效:
```bash
source ~/.bashrc
```
#### 测试CUDA是否正常运作
最后一步就是验证整个流程下来设置无误与否了。可以参照官方给出的例子来进行简单的功能检测。比如编写一段小程序来查看当前可用设备列表及其基本信息等[^2]:
```c
#include <cuda_runtime.h>
#include <stdio.h>
int main() {
int nDevices;
cudaGetDeviceCount(&nDevices);
printf("Number of devices: %d\n", nDevices);
for(int i=0; i<nDevices; ++i){
cudaDeviceProp prop;
cudaGetDeviceProperties(&prop, i);
printf("\n Device Number:%d \n", i);
printf(" Device name: %s\n", prop.name);
printf(" Memory Clock Rate(KHz):%d\n", prop.memoryClockRate);
printf(" Memory Bus Width(bits):%d\n", prop.memoryBusWidth);
printf(" Peak Memory Bandwidth(GB/s):%.2f\n",
2.0*prop.memoryClockRate*(prop.memoryBusWidth/8)/1e6);
}
}
```
保存上述源码到名为 test.cu 的文件当中;接着利用nvcc编译器将其转换成二进制形式的应用程序;
```bash
nvcc test.cu -o test_cuda
./test_cuda
```
如果一切顺利的话,则应该可以看到关于本机所连接图形处理器的相关参数输出结果。
阅读全文
相关推荐


















