news 2026/9/8 17:41:49

WSL2配置CUDA+conda+PyTorch

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
WSL2配置CUDA+conda+PyTorch

前言

在 WSL2 中搭建深度学习环境,是许多开发者高效开展 AI 研究的常见选择。本文聚焦 CUDA、conda 与 PyTorch 的配置流程,重点讲解如何从官网自主获取资源并完成环境搭建,助你快速掌握核心步骤,轻松开启深度学习实践。本文我将用 Ubuntu24.04 进行演示,其中的配置思路是可以迁移的。

一、换源

在 WSL2 中安装软件前,建议先将 apt 源更换为国内镜像源,以大幅提升下载速度。具体操作如下:

1. 更新软件列表
首先执行更新命令,确保本地包索引是最新的:

sudo apt update

2. 备份原始源文件

为了防止配置出错,建议先备份原始的 sources 文件

sudo cp /etc/apt/sources.list.d/ubuntu.sources /etc/apt/sources.list.d/ubuntu.sources.bak

3. 编辑源配置文件
使用 vim 编辑器打开源配置文件进行修改(如果你不习惯 vim,也可以使用 nano):

sudo vim /etc/apt/sources.list.d/ubuntu.sources
  • 清空旧内容:进入编辑器后,这里需要按“i”进行插入模式,再输入:%d,这将删除文件内的所有内容。
  • 粘贴新配置:将以下内容复制并粘贴到文件中(这里使用的是清华大学的镜像源)
Types: deb URIs: https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ Suites: noble noble-updates noble-security noble-backports Components: main restricted universe multiverse Signed-By: /usr/share/keyrings/ubuntu-archive-keyring.gpg

如图:

  • 保存并退出:按ESC键退出插入模式,输入:wq并回车,即可保存文件并退出编辑器。
  • 验证修改:你可以使用以下命令查看文件内容,确认修改是否成功
cat /etc/apt/sources.list.d/ubuntu.sources

如图:

4.更新软件源索引:

配置完成后,务必执行更新命令,让系统重新读取新的源地址:

sudo apt update

二、配置 CUDA

1. 安装依赖

在 WSL2 的 Linux 环境中,我们需要安装 NVIDIA 的开源驱动包作为依赖(注意:这通常是为了配合 Windows 宿主机的驱动使用):

sudo apt install nvidia-open

2.重启系统

安装完成后,为了确保驱动加载生效,需要重启 WSL2 或整个系统:

sudo reboot

3. 检查状态

重启后,输入以下命令查看显卡状态及 CUDA 版本信息检查

nvidia-smi

这里的 CUDA Version:显示了当前驱动支持的最高 CUDA 版本,之后需要根据这个去安装 CUDA Toolkit

4. 安装 CUDA Toolkit

点击打开官网,根据一下步骤打开 CUDA Toolkit

如图这里需要根据CUDA Version的情况下载对应的版本,如果在 13.3 一下就需要去找其它版本的 CUDA Toolkit。

这里大家就根据自己的 CUDA Version 选择对应的 Toolkit 下载即可。

这里因为我的 CUDA Version 是 13.2,我就以 13.2 作为演示。

这里根据指示选择好要下载的类型,注意 Installer Type 是可以三选一的,我这里用 deb 做演示,如图:

选择好之后下面会出现一些命令行,可以直接复制粘贴到终端中运行,如图。

5.配置环境变量

下载并安装好 CUDA Toolkit 后,系统通常不会自动将其添加到路径中。我们需要手动配置环境变量,以便系统能找到 CUDA 的编译器和库文件。

  • 确认安装目录:首先,我们可以查看/usr/local目录下是否存在对应的 CUDA 文件夹,如图:
ls /usr/local

接着查看该目录下的具体文件结构,如图:

ls /usr/local/cuda-13.2

  • 写入配置文件:
    使用以下命令将 CUDA 的路径追加到~/.bashrc配置文件中。这里使用了cat命令配合 Here Document (<<'EOF') 的方式一次性写入两行关键配置:
cat >> ~/.bashrc <<'EOF' export PATH=/usr/local/cuda-13.2/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-13.2/lib64:$LD_LIBRARY_PATH EOF

如图:

注意: 请确保将命令中的13.2替换为你实际安装的 CUDA 版本号。

  • 使配置生效:修改完配置文件后,需要执行以下命令使其立即生效
source ~/.bashrc

6. 验证安装

最后,通过查看 NVCC(NVIDIA CUDA Compiler)的版本号来验证环境是否配置成功:

nvcc --version

如果输出了类似图中的版本信息,说明 CUDA Toolkit 已经安装并配置成功,你可以开始进行深度学习开发了。

三、安装 conda

1. 获取安装脚本

打开官网,我们选择跳过登录就可以进行下载

选择安装 Linux 版本,这里推荐安装 Miniconda,因为它更加轻量、干净,我们可以点击下面的 Linux install instructions

这里会跳转到下载Miniconda 的文档,想要了解更多信息的可以自行学习,这里我把关键的命令行给出。

这里我们下载最新的 Linux 64-Bit 安装脚本:

curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh

2. 运行安装脚本

下载完成后,执行以下命令开始安装:

bash ~/Miniconda3-latest-Linux-x86_64.sh

3. 配置环境变量

安装完成后,为了确保系统能随时识别conda命令,我们需要将 Miniconda 的路径添加到系统的环境变量中。

使用以下命令将路径配置写入~/.bashrc文件:

cat >> ~/.bashrc <<'EOF' export PATH=$HOME/miniconda3/condabin:$PATH EOF

接着,执行以下命令使刚才的配置立即生效:

source ~/.bashrc

4. 查看环境

配置生效后,我们可以通过以下命令检查 Conda 是否正常工作,并查看当前的环境列表:

conda env list

如果安装成功,终端会输出如图的信息:

5. 验证激活状态

此时,如果你重新启动终端,或者看到命令行提示符的最前方出现了(base)字样,如图所示,就说明 Conda 已经成功激活并默认进入了基础环境。

四、安装 PyTorch

1. 创建虚拟环境

先创建一个名为deeplearning的独立环境,并指定 Python 版本为 3.11(以根据需要修改版本号),同时安装pip包管理工具(后面需要用这个来安装 PyTorch):

conda create -y -n deeplearning python=3.11 pip

2. 激活环境

环境创建完成后,使用以下命令激活它:

conda activate deeplearning

激活成功后,你的命令行提示符前方会出现(deeplearning)字样,表示当前操作都在该隔离环境中进行。

3.获取安装命令

  • 访问官网,点击“Get started”按钮,进入安装配置页面

  • 这里按照提醒选择对应的配置,注意 CUDA 的版本,要跟前面下载的 CUDA 一致(因为我前面下载的是 13.2,我这里就下载 13.2)。

  • 选择完毕后,会自动生成对应的安装命令。直接复制该命令到终端中运行。

4. 验证结果

在终端中输入以下命令,列出当前环境中所有已安装的 Python 包:

pip list

如果安装成功,会显示类似如图的版本。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/9/8 17:41:48

mbed OS源码架构解析:HAL、RTOS与驱动层设计

1. 从“点灯都要重新造轮子”说起&#xff1a;mbed OS 到底解决了什么问题 做 Arm 嵌入式开发的兄弟应该都有过这种经历&#xff1a;芯片换了一颗&#xff0c;板子接口完全不一样&#xff0c;以前写的 GPIO 初始化代码全废了&#xff0c;打开新芯片的参考手册从头翻寄存器。明明…

作者头像 李华
网站建设 2026/9/8 17:40:00

多智能体协作课堂化:OpenMAIC四种交互模式与MCP集成实践

第一次跑多智能体项目的时候&#xff0c;我天真地以为把几个大模型拉进同一个群里就完事了。结果三个Agent聊了不到五轮&#xff0c;就开始互相客套&#xff1a;“你说得很对”“我补充一点”“同意楼上”&#xff0c;然后彻底忘了最初要解决什么问题。那一刻我突然意识到&…

作者头像 李华
网站建设 2026/9/8 17:39:08

SL-WX-Captcha 行为验证组件(三合一)

SL-WX-Captcha 行为验证组件&#xff08;三合一&#xff09; 插件市场 操作视频 组件简介 单一组件统一支持三种主流前端行为验证方式&#xff0c;通过 mode 属性零切换&#xff1a; slide 滑动验证&#xff1a;经典滑块&#xff0c;按住并拖动到底部最右侧即视为通过&#…

作者头像 李华