admin 管理员组

文章数量: 887021

(补充:Windows11用户请安装WSL 2,WSL 1是存在问题的)

本文假设已成功安装ubuntu子系统。首先声明,cuda与cudnn在子系统内部安装,显卡驱动需要在windows10安装,而子系统内部不需要安装显卡驱动。如果在子系统内部执行命令:如果nvidia-smi能够正确显示显卡当前状态,那么恭喜,你的系统版本满足要求,此时跳过步骤一,直接进行步骤二即可;nvidia-smi 无法显示显卡当前状态,那么说明需要执行下述步骤一。

一、Windows10版本

Windows10需要21H2版本及以上(否则无法在子系统内部安装cuda)

查看windows10版本方法:

win+r键,输入命令: winver,得到如下图所示:

1.1 将windows10更新至21H2版本的步骤如下:

1)首先到windows10的官网下载更新助手软件Windows10Upgrade9252.exe,官网地址:

https://go.microsoft/fwlink/?LinkID=799445

2)运行Windows10Upgrade9252.exe,并一路点击下一步(没什么可选的选项,等待安装就行了,后续会要求重启电脑进行安装,点击立即重启即可)

二、NVIDIA显卡驱动

显卡驱动在windows10系统中安装,使用Nvidia GeForce experience即可,此处不再赘述。

三、CUDA与CUDNN安装

本文使用CUDA==11.3.1与CUDNN==8.2.1

3.1 WSL子系统内CUDA安装

3.1.1 如果选择其他版本CUDA与CUDNN,参考CUDA官网历史版本网址:CUDA Toolkit Archive | NVIDIA Developer

如图中所示,选择WSL-Ubuntu版本,根据命令安装即可。 

3.1.2 CUDA11.3安装

        1)官网的安装命令如下:

wget https://developer.download.nvidia/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run

sudo sh cuda_11.3.1_465.19.01_linux.run

        2)安装依赖:

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev 

        3)CUDA环境变量:

sudo vim ~/.bashrc

在结尾添加:

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

 更新文件:

source ~/.bashrc

至此,CUDA安装完成。

CUDNN安装参考如下链接3.2节:Windows 11/10 WSL2 Ubuntu 20.04 下配置Cuda及Pytorch_Quatttschen的博客-CSDN博客_wsl2安装cudnn

此处温馨提示:windows的各个分区挂载于 /mnt/ 目录下,比如d盘路径为 /mnt/d ,因此不必一定在WSL系统内部下载相关软件包,也可在windows下载后直接在ubuntu内解压并运行。

四、SSH访问WSL系统

1. 配置并安装SSH服务启动

sudo apt install openssh-server

安装完成后,修改文件 /etc/ssh/sshd_config,增加或修改如下2行(在初始文件中是注释状态):

Port 2222

PasswordAuthentication yes

2. 删除/run/nologin目录

sudo rm -rf /run/nologin

3.通过用户名和ip访问WSL

ssh name@ip_addr -p 2222

至此完成。

本文标签: 子系统 WSL Ubuntu CUDNN CUDA