美文网首页
wsl2配置GPU CUDA 进行神经网络训练 预测

wsl2配置GPU CUDA 进行神经网络训练 预测

作者: 夏过的雨 | 来源:发表于2020-07-29 12:16 被阅读0次

为什么要在wsl2中配置GPU Ubuntu系统不香吗?

我这个人比较喜欢折腾整天装个这软件那软件的Ubuntu就不干了啊动不动给我罢工啊 重新安装个显卡驱动啥的都是小事情关键是经常修改完成之后启动不了 Ubuntu折腾累 现在wsl2可以使用GPU 天啊 window10这么香了吗?

必须重装系统挂载window10 EXT3硬盘 一切搞定

首先我们需要一个配置好的wsl2

Ubuntu-18.04

我使用的是Ubuntu-18.04

我已经预先安装了docker 可以正常使用

下面我们开始安装cuda驱动

选择驱动类型

当然这里不需要选择操作系统了

下载安装就可以了 这里安装的是windows 驱动它提供了wsl2的驱动所以说 wsl2中的Ubuntu系统或者其他系统都是不需要安装显卡驱动

接下来就是官方文档

https://docs.nvidia.com/cuda/wsl-user-guide/index.html#installing-nvidia-drivers

官方文档首先说的是 你的Linux子系统 内核版本必须高于或等于 4.19.121

当然系统版本也是有要求的比如我的 内部版本 20175

winver命令查看

这是测试版本

https://www.jianshu.com/p/aa52b39bae64

更新方式如上

接受Linux子系统内核更新必须打开这个选项

接受其他Microsoft产品更新

当然你也可以自己编译内核然后更新

这一切都准备好了 我们就可以进入wsl的Ubuntu中了

apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub

 sh -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

 sh -c 'echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/cuda.list'

 apt-get update

有些网络不能访问developer.download.nvidia.com 请换个网络访问 比如手机网络

 apt-get update

安装apt-get install -y cuda-toolkit-11-0 还是那句话不要安装显卡驱动所以不要选择安装cuda, cuda-11-0, or cuda-drivers

这个比较大下载慢耐心等待

下载安装完成后 切换到 /usr/local/cuda/samples 目录

只执行 make 编译

测试通过

相关文章

网友评论

      本文标题:wsl2配置GPU CUDA 进行神经网络训练 预测

      本文链接:https://www.haomeiwen.com/subject/fikhrktx.html