【AI】搭建Windows Linux子系统(WSL2)CUDA环境
创始人
2025-01-08 15:34:40
0

0.准备工作

Windows本机安装CUDA Driver
首先去下载页面下载驱动文件
在这里插入图片描述

点击Get CUDA Driver进入下载页面,我看下载页面跟普通驱动下载页面相同,感觉应该不是单独的驱动,只要之前显卡已经安装好了CUDA的驱动,就可以先省略这一步。

1.安装Windows子系统

win11中安装wsl子系统非常方便,只需要执行一个命令即可,系统会自动安装Ubuntu系统,执行完之后需要重启电脑,重启后需要设置用户名和密码,根据自己实际情况设置即可。

wsl --install 

安装完重启电脑设置完账号密码后,会看到如下页面:
在这里插入图片描述
这时候就可以操作linux系统了

2.WSL安装CUDA Toolkit

官方的教程地址:https://docs.nvidia.com/cuda/wsl-user-guide/index.html
进入官方下载地址:https://developer.nvidia.com/cuda-toolkit-archive
根据自己安装的CUDA驱动版本选择CUDA Toolkit版本,这里最好与主机的驱动版本一致,否则不知道会出现什么奇怪错误。

在这里插入图片描述
去wsl命令行执行wget命令,可以直接在linux系统上下载Toolkit,然后执行安装命令

wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run sudo sh cuda_12.2.0_535.54.03_linux.run 

安装时可能会报错:Failed to verify gcc version. See log at /var/log/cuda-installer.log for details.
这是因为我们的wsl系统刚刚创建,可能没有安装gcc导致的,可以参考这篇文章解决。
这里也贴一下吧

sudo apt update && sudo apt upgrade -y sudo apt install build-essential # 查看安装的gcc版本 gcc --version 

然后再执行上面的安装程序,安装完之后打印了下面的信息

=========== = Summary = ===========  Driver:   Not Selected Toolkit:  Installed in /usr/local/cuda-12.2/  Please make sure that  -   PATH includes /usr/local/cuda-12.2/bin  -   LD_LIBRARY_PATH includes /usr/local/cuda-12.2/lib64, or, add /usr/local/cuda-12.2/lib64 to /etc/ld.so.conf and run ldconfig as root  To uninstall the CUDA Toolkit, run cuda-uninstaller in /usr/local/cuda-12.2/bin ***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 535.00 is required for CUDA 12.2 functionality to work. To install the driver using this installer, run the following command, replacing  with the name of this run file:     sudo .run --silent --driver  Logfile is /var/log/cuda-installer.log 

这里的warning可以不用处理,官方文档上说不需要安装driver

安装完成之后需要将路径加入到环境变量中:

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

3.安装cuDNN(可选)

NVIDIA CUDA® 深度神经网络库 (cuDNN) 是一个 GPU 加速的深度神经网络基元库,能够以高度优化的方式实现标准例程(如前向和反向卷积、池化层、归一化和激活层)。

全球的深度学习研究人员和框架开发者都依赖 cuDNN 来实现高性能 GPU 加速。借助 cuDNN,研究人员和开发者可以专注于训练神经网络及开发软件应用,而不必花时间进行低层级的 GPU 性能调整。cuDNN 可加速广泛应用的深度学习框架,包括 Caffe2、Chainer、Keras、MATLAB、MxNet、PaddlePaddle、PyTorch 和 TensorFlow。如需获取经 NVIDIA 优化且已在框架中集成 cuDNN 的深度学习框架容器,请访问 NVIDIA GPU CLOUD 了解详情并开始使用。

下载地址:https://developer.nvidia.cn/cudnn 或 https://developer.nvidia.com/rdp/cudnn-archive#a-collapse896-120

在运行PyTorch训练和推理时,通常需要安装cuDNN,但并非绝对必要。cuDNN是一个GPU加速的深度神经网络库,能够以高度优化的方式实现标准例程,如前向和反向卷积、池化层、归一化和激活层。

在某些情况下,部分深度学习框架会将CUDA、CUDNN打包在whl包里,因此无需手动安装。例如,PyTorch的大部分版本就是这样,所以无需手动安装。

4.修改WSL安装路径

显然,此时的wsl默认安装在c盘,随着系统的使用,会占用我们C盘的空间,所以我们将其打包放到其它盘去。
1.查看WSL发行版本
在Windows PowerShell中输入命令:
wsl -l --all -v

PS C:\Users\Admin> wsl -l --all -v   NAME      STATE           VERSION * Ubuntu    Stopped         2 

2.导出分发版为tar文件到d盘
wsl --export Ubuntu d:\wsl-ubuntu20.04.tar(Ubuntu修改成你现在的发行版名称)
3.注销当前分发版
wsl --unregister Ubuntu (Ubuntu修改成你现在的发行版名称)
4.重新导入并安装WSL在d:\wsl-ubuntu20.04(可以修改成你自己想要的目录)
wsl --import Ubuntu d:\wsl-ubuntu20.04 d:\wsl-ubuntu20.04.tar --version 2
5.设置默认登陆用户为安装时用户名
Ubuntu config --default-user Username
6.删除tar文件(可选)
del d:\wsl-ubuntu20.04.tar

参考文档

5.WSL与主机传输文件

从windows拷贝文件到WSL
先查看目录挂载情况
sudo ls /mnt/*
找到文件,拷贝到当前目录
cp /mnt/盘符/文件 .

附WSL官方文档地址:https://learn.microsoft.com/zh-cn/windows/wsl/

参考文章:https://zhuanlan.zhihu.com/p/436393852

相关内容

热门资讯

推荐练习!广西友乐软件辅助,微... 推荐练习!广西友乐软件辅助,微信微乐辅助器免费安装,详细教程(有挂规律)小薇(辅助器软件下载)致您一...
详情窍门!老友广东潮汕麻雀辅助... 详情窍门!老友广东潮汕麻雀辅助,樱花之盛辅助器,详细教程(有挂实锤);运老友广东潮汕麻雀辅助辅助工具...
有挂方式!边锋斗地主插件辅助脚... 有挂方式!边锋斗地主插件辅助脚本,欢乐达人暗堡破解,详细教程(发现有挂)1、许多玩家不知道边锋斗地主...
关于方针!陕麻圈辅助开挂软件,... 关于方针!陕麻圈辅助开挂软件,小闲川南辅助工具,详细教程(有挂秘笈)1、在陕麻圈辅助开挂软件插件功能...
揭幕方式!赣湘互娱透视挂,神兽... 揭幕方式!赣湘互娱透视挂,神兽互娱辅助免费,详细教程(存在有挂)1、神兽互娱辅助免费免费辅助多个强度...
开挂举措!广东雀神麻雀控制器,... 开挂举措!广东雀神麻雀控制器,老友广东辅助器,详细教程(有挂解密)广东雀神麻雀控制器脚本下载中分为三...
关于攻略!多乐找刺激捕鱼辅助,... 您好,多乐找刺激捕鱼辅助这款游戏可以开挂的,确实是有挂的,需要了解加去威信【485275054】很多...
了解教材!九神大厅辅助,全民比... 了解教材!九神大厅辅助,全民比鸡bug,详细教程(有挂教学)所有人都在同一条线上,像星星一样排成一排...
解谜演示!微信小程序有挂吗辅助... 解谜演示!微信小程序有挂吗辅助,hhpoker辅助器,详细教程(确实有挂)1、用户打开应用后不用登录...
了解模块!途游小程序辅助器,广... 了解模块!途游小程序辅助器,广西友乐辅助app,详细教程(真实有挂)1、途游小程序辅助器免费脚本咨询...