Win11+WSL2安装cuda11.8，Anaconda配置Pytorch2.4-GPU版本

最新推荐文章于 2025-06-15 15:58:48 发布

月光下的洪流

最新推荐文章于 2025-06-15 15:58:48 发布

阅读量1.3k

点赞数 20

CC 4.0 BY-SA版权

文章标签： linux 运维服务器

1.WSL安装

首先安装WSL2，按照这个教程，目前我是安装到E盘

安装完之后，输入nvidia-smi，会显示Command ‘nvidia-smi‘ not found, but can be installed with，按照下面的教程解决（亲测好使）

然后就可以看到自己的显卡型号，以及支持的最高的CUDA版本

进入 ~下载CUDA ，下面链接为CUDA11.8下载的链接，直接按照命令下载就可以

在WSL2的终端中，使用以下命令安装GCC和编译所需的工具：

sudo apt-get update
sudo apt-get install build-essential

安装完成后，使用以下命令查看GCC的版本，以确保安装成功：

gcc --version

如下图所示表示安装成功

在安装GCC之后，您可以继续进行CUDA 11.8的安装。使用您之前下载的CUDA安装文件：

sudo sh cuda_11.8.0_520.61.05_linux.run

在安装过程中，按照提示操作，输入accept继续安装。如下图显示安装成功

安装完成后，将CUDA的路径添加到环境变量中。编辑~/.bashrc文件，添加以下内容：

输入

vim ~/.bashrc

进入编辑，输入i，进入编辑模式，输入下面的内容

export PATH=$PATH:/usr/local/cuda-11.8/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64

显示如下

然后按下ESC键，输入 :wq，退出编辑

运行以下命令：

nvcc --version

如下图所示，显示安装成功

使用清华镜像源下载

输入命令进行安装,bash 后面的是下载的Anaconda的版本

bash Anaconda3-xxxxx

安装过程一路输入yes。

下图显示安装成功

配置环境变量：安装完成后，将 Anaconda 的路径添加到环境变量中。编辑~/.bashrc文件，添加以下内容：

export PATH=$PATH:/home/用户名/anaconda3/bin

保存文件后，运行以下命令使更改生效：

source ~/.bashrc

验证安装是否成功：

conda --version

如下图所示，显示安装成功

conda create -n torch2.4 python=3.11

激活环境

conda activate pytorch2.4

找到pytorch官网Previous PyTorch Versions | PyTorch，输入默认的安装命令

conda install pytorch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0  pytorch-cuda=11.8 -c pytorch -c nvidia