在WSL中使用GPU:WSL2 + Ubuntu 18.04 + CUDA + Gnome图形界面环境配置


引言

鉴于在Linux环境下运行深度学习代码,并使用PyCharm调试的需求,原本打算安装Ubuntu双系统,但了解到WSL2已经支持使用GPU进行CUDA运算,故考虑使用WSL2实现上述需求。本文记录了在Windows 10下安装WSL2,并配置Gnome图形界面和CUDA的过程。


1. 确认Windows 10版本

需要注意的是WSL2和CUDA Toolkit对Windows的版本要求并不是完全相同的。即便能够成功安装WSL2,也不一定可以成功使用CUDA。

根据微软官方文档的要求,在x64机器上安装WSL2需要Windows版本 1903 或以上,内部版本 18362 或以上。
在这里插入图片描述

而根据NVIDIA官方文档的要求,安装适用于WSL2的CUDA Toolkit,需要安装Windows Insider Preview版本并切换到Fast Ring通道。版本要求20145或以上,但对家庭版/专业版则没有要求。
在这里插入图片描述

总体来看只要达到了安装CUDA on WSL的版本要求,就达到了安装WSL2的要求

2. 在Windows上安装WSL2

这一步参考微软官方文档完成。

3. 在Windows上安装CUDA on WSL驱动

在WSL2中使用GPU,并不需要在Linux内安装驱动,而是需要在Windows中安装特定版本的驱动。

评论 25
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值