要求: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的下载:
打开网址之后会出现如下的界面,读者可以选择合适自己的版本进行下载:
进入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安装(必须选择合适版本)
点击下面链接:
初次下载时,可能需要注册信息,按照步骤提示注册即可。
注册完毕之后,选择适合自己安装的cuda版本的cudnn进行下载,将下载之后的文件压缩包进行解压即可,将解压后的文件放入之前建立的Ubuntu文件夹中。
找到解压之后的文件夹路径,进入该文件夹里,输入以下命令即可:
sudo cp lib/* /usr/local/cuda/lib64/
sudo cp include/* /usr/local/cuda/include/
如下图所示:
3.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”,则说明环境配置好了。