WSL安装Ubuntu和GPU版的Pytorch的安装

要求:Windows10版本2004及更高版本(内部版本19041及更高版本)或Windows11。这里最好使用Windows10专业版和Windows11专业版。

一、WSL安装Ubuntu。

首先打开电脑属性,界面如下,查看当前Windows版本是否满足要求,不满足则需要将Windows版本升级。

在电脑本地启动wsl功能

在开始菜单输入控制面板,点击进入,然后点击程序

点击“启用或关闭Windows功能”

首先启动Hype-V服务

然后勾选wsl服务

以上操作都做完之后,将电脑重启更新

打开Microsoft Store,在搜索框里输入“Ubuntu”,点击搜索。

选择需要安装的Ubuntu版本,这里建议推荐Ubuntu 20.04.5 LTS和Ubuntu 18.04.5 LTS这两个版本,选择一个即可。

打开Windows PowerShell(以管理员身份运行打开),在命令行中输入wsl --update

以上命令执行完毕之后,重新返回Microsoft Store,打开Ubuntu系统。

点击“打开”按钮之后,会显示如下界面:

用户只需输入用户名和密码即可登录。

注:下面这一步骤可以省略

登录之后可以再使用Windows PowerShell(以管理员身份运行打开),查看到当前的wsl linux的状态,其中包括子系统的名称、状态和版本。如下图所示:

二、GPU版的Pytorch的安装

按照上面的步骤安装Ubuntu之后,返回Ubuntu界面,输入nvidia-smi命令,出现如下界面,说明WSL的显卡驱动安装完毕:

在C盘或者D盘创建一个Ubuntu的文件夹,用于存放下载的安装包。

1.cuda安装

下面进行cuda的安装,打开下面的网址,进行cuda的下载:

CUDA Toolkit Archive | NVIDIA Developer

打开网址之后会出现如下的界面,读者可以选择合适自己的版本进行下载:

进入Ubuntu文件夹中,将以下指令输入Ubuntu界面中

wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
sudo sh cuda_12.1.0_530.30.02_linux.run

输入第一个指令,进行cuda的下载,如下图所示。

输入第二个指令之后,可能会出现如下图所示的情况:

在遇到上面的这个情况的时候,只需在指令的后面添加--override即可,如下图所示。

输入sudo sh cuda_11.2.0_460.27.04_linux.run  --override,输入该指令之后等待半分钟到一分钟左右,会出现下图界面,读者只需要输入“accept”即可

按方向键选择install进行cuda安装:

出现如下界面,则说明安装完毕且成功:

配置环境变量

输入vim ~/.bashrc命令打开文件,在文件结尾输入以下语句

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

更新环境变量配置 source ~/.bashrc

当输入nvcc -V指令,出现如下图所示,则说明cuda安装成功:

2.cudnn安装(必须选择合适版本)

点击下面链接:

cuDNN Archive | NVIDIA Developer

初次下载时,可能需要注册信息,按照步骤提示注册即可。

注册完毕之后,选择适合自己安装的cuda版本的cudnn进行下载,将下载之后的文件压缩包进行解压即可,将解压后的文件放入之前建立的Ubuntu文件夹中。

找到解压之后的文件夹路径,进入该文件夹里,输入以下命令即可:

sudo cp lib/* /usr/local/cuda/lib64/ 
sudo cp include/* /usr/local/cuda/include/

如下图所示:

3.anaconda下载

打开官网链接:

Free Download | Anaconda

选择合适的Linux版本进行下载:

下载完毕之后将文件存放入Ubuntu文件夹中,输入下面指令进行安装:

sudo sh Anaconda3-2022.10-Linux-x86_64.sh

出现如下图所示,即安装成功:

配置环境变量

在终端输入命令

vim ~/.bashrc

在打开文件最后一行添加:

export PATH="/home/lmh/anaconda3/bin:$PATH"

这里的路径选择是你的Anaconda的默认安装路径,默认在home下。点击保存,关闭bashrc文件。

最后更新环境变量配置:

source ~/.bashrc

配置完conda的环境变量之后,输入conda命令,出现如下图所示,说明conda安装成功。

4.pytorch安装

修改权限:

sudo chmod a+w .conda

创建虚拟环境:

conda create -n pytorch-gpu python=3.8

查看创建的虚拟环境:

conda env list

进入虚拟环境:

 Conda activate torch-gpu

点击官网链接:

https://pytorch.org/get-started/locally/

 选择合适的pytorch版本进行安装:

输入命令:

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

 5.验证环境

当输入下面命令是,如果出现“True”,则说明环境配置好了。

  • 8
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值