当前位置:首页 > 系统教程 > 正文

WSL2-Ubuntu深度学习环境搭建(Anaconda、CUDA13.0、cuDNN9.12及PyTorch安装与验证教程)

WSL2-Ubuntu深度学习环境搭建(Anaconda、CUDA13.0、cuDNN9.12及PyTorch安装与验证教程)

本教程将详细介绍在WSL2-Ubuntu系统中安装Anaconda、CUDA13.0、cuDNN9.12及PyTorch的完整步骤,并提供环境验证方法。通过本指南,即使您是初学者,也能轻松搭建深度学习环境,为后续AI项目打下基础。我们将从先决条件开始,逐步安装每个组件,确保过程清晰易懂。

WSL2-Ubuntu深度学习环境搭建(Anaconda、CUDA13.0、cuDNN9.12及PyTorch安装与验证教程) WSL2 Ubuntu  Anaconda CUDA 13.0 PyTorch 第1张

1. 先决条件:确保WSL2和Ubuntu已安装

在开始之前,请确认您的Windows系统已启用WSL2,并安装了Ubuntu发行版(建议Ubuntu 20.04或更高版本)。打开PowerShell,以管理员身份运行以下命令来检查WSL状态:

    wsl --list --verbose  

如果未安装,请参考微软官方文档安装WSL2和Ubuntu。本教程假设您已熟悉基本Linux命令,并在WSL2-Ubuntu终端中操作。

2. 安装Anaconda:Python环境管理

Anaconda是流行的Python发行版,能简化包管理和环境控制。首先,从Anaconda官网下载Linux安装脚本,或使用wget命令在终端中下载:

    wget https://repo.anaconda.com/archive/Anaconda3-2023.03-Linux-x86_64.sh  

然后,运行安装脚本并按照提示完成Anaconda安装。安装后,重启终端或运行source ~/.bashrc来激活conda命令。验证Anaconda安装是否成功:

    conda --version  

如果显示版本号,则Anaconda安装完成。接下来,我们将安装CUDA 13.0,这是NVIDIA的并行计算平台,对深度学习至关重要。

3. 安装CUDA 13.0:GPU加速支持

在WSL2-Ubuntu中安装CUDA 13.0需要先确保系统有NVIDIA GPU驱动(在Windows主机中安装)。然后,访问NVIDIA官网下载CUDA 13.0安装包,或使用终端命令安装。首先,添加NVIDIA仓库并更新:

    wget https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/cuda-wsl-ubuntu.pinsudo mv cuda-wsl-ubuntu.pin /etc/apt/preferences.d/cuda-repository-pin-600sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/3bf863cc.pubsudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/wsl-ubuntu/x86_64/ /"sudo apt-get update  

然后,安装CUDA 13.0:

    sudo apt-get -y install cuda-13-0  

安装完成后,添加环境变量到~/.bashrc文件:export PATH=/usr/local/cuda-13.0/bin${PATH:+:${PATH}}export LD_LIBRARY_PATH=/usr/local/cuda-13.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}。运行source ~/.bashrc后,验证CUDA 13.0安装:

    nvcc --version  

如果输出CUDA版本信息,则安装成功。接下来,安装cuDNN 9.12,这是针对深度神经网络的GPU加速库。

4. 安装cuDNN 9.12:深度学习库优化

cuDNN需要从NVIDIA官网下载(需注册账户)。下载适用于CUDA 13.0的cuDNN 9.12 Linux版本。在终端中,使用tar命令解压并复制文件:

    tar -xzvf cudnn-13.0-linux-x64-v9.12.0.tgzsudo cp cuda/include/cudnn*.h /usr/local/cuda-13.0/includesudo cp cuda/lib64/libcudnn* /usr/local/cuda-13.0/lib64sudo chmod a+r /usr/local/cuda-13.0/include/cudnn*.h /usr/local/cuda-13.0/lib64/libcudnn*  

完成后,验证cuDNN安装是否成功,可以检查文件是否存在。现在,我们将安装PyTorch,这是一个流行的深度学习框架,并利用已安装的CUDA和cuDNN。

5. 安装PyTorch:深度学习框架

使用Anaconda环境安装PyTorch可以确保依赖兼容。首先,创建一个新的conda环境(可选):

    conda create -n pytorch_env python=3.9conda activate pytorch_env  

然后,根据PyTorch官网命令安装支持CUDA 13.0的版本。运行以下命令:

    conda install pytorch torchvision torchaudio pytorch-cuda=13.0 -c pytorch -c nvidia  

安装过程可能需要几分钟。完成后,我们将进行环境验证,确保所有组件正常工作。

6. 完整环境验证:测试PyTorch和CUDA

运行Python代码来验证PyTorch是否成功识别CUDA 13.0和cuDNN 9.12。在终端中启动Python:

    python  

在Python交互环境中,输入以下代码:

    import torchprint("PyTorch版本:", torch.version)print("CUDA是否可用:", torch.cuda.is_available())print("CUDA版本:", torch.version.cuda)print("cuDNN版本:", torch.backends.cudnn.version())if torch.cuda.is_available():    device = torch.device("cuda")    print("当前GPU设备:", torch.cuda.get_device_name(0))else:    print("CUDA不可用,请检查安装。")  

如果输出显示CUDA可用,并正确显示版本号,则环境搭建成功。这表明您已在WSL2-Ubuntu中完整安装了Anaconda、CUDA13.0、cuDNN9.12及PyTorch。

7. 总结与后续步骤

本教程逐步引导您在WSL2-Ubuntu系统中安装深度学习环境。通过使用Anaconda管理Python环境,安装CUDA 13.0和cuDNN 9.12以启用GPU加速,并安装PyTorch进行深度学习开发,您已经为AI项目做好了准备。如果遇到问题,请检查驱动版本或参考官方文档。现在,您可以开始使用PyTorch进行模型训练和实验了!

本教程关键词:WSL2 Ubuntu, Anaconda, CUDA 13.0, PyTorch。这些技术是构建深度学习环境的核心,确保在文章中多次提及以优化SEO。