Windows10/11 WSL2 安装nvidia

您所在的位置:网站首页 英伟达驱动下载安装 Windows10/11 WSL2 安装nvidia

Windows10/11 WSL2 安装nvidia

2023-11-12 11:40| 来源: 网络整理| 查看: 265

很多人反映驱动地址没有了,我点进去看了一下,确实是这样。原因是:以前带有wsl2驱动的地址是“过渡时期”的过渡方案,类似于预览版:

现在英伟达好像把wsl2驱动集成到正式版的驱动上了,下面是英伟达的wsl介绍网址:

点进去后,再点击get cuda就直接跳到驱动下载网页了

所以,有可能直接在这个驱动下载地址下载对应的驱动即可。

此外,兄弟们一定要记得下载最新的Windows10,有时候在设置中检查更新时,虽然已经提示是最新版本了,但其实版本号还没有达到标准。我的建议是,有条件的直接升级到win11。

驱动安装背景

最近在实验室的Windows服务器上学习“炼丹”知识,发现速度不尽人意,即使设置了使用GPU训练,但总觉得Windows没有压榨出显卡全部实力。本科的时候听说过WSL,其实就是Windows下的Linux子系统,我记得第一代的WSL不能安装显卡驱动,毛病挺多的……最近WSL2出山,听很多人说不错,也可以安装驱动。折腾的过程中踩了很多坑,希望我前几天的经验能够帮助到你们。

服务器的硬件情况

Intel(R) Core(TM) i9-10940X CPU @ 3.30GHz   3.31 GHz

NVIDIA GeForce RTX 3090 * 3

Windows10系统(系统版本是很大的坑!一会会讲)

开始安装

1. 安装WSL2

这一步可以参考Windows官网,以及其他B站UP主的安装教程,贴两个链接(一定要装WSL2)。

2. 安装Windows的nvidia驱动(Windows下操作)

下载并安装带有WSL2驱动的nvidia-Windows驱动

根据自己电脑显卡类型,分GeForce和QUADRO两个版本,自己电脑是哪个就选哪个。下载驱动后,安装。(安装过程中可以直接选择“精简”模式)

带有WSL驱动的官方驱动下载地址:https://developer.nvidia.cn/cuda/wsl/download

注意!

有人会问:不是安装WSL2的驱动吗?为什么装Windows的?实际上,官网文档上标注 了,带有WSL2的官方nvidia驱动是整个过程唯一要装的GPU驱动!!!千万别被其他指南带错路了!(我就是没找到这个,白白浪费很多时间)

Nvidia-WSL驱动官方文档:https://docs.nvidia.cn/cuda/wsl-user-guide/index.html

检测是否安装成功

cmd命令行输入:

如果出现类似于下图效果,即安装完成。

检测驱动是否安装完成

记下图中的CUDA version

3. 在WSL2中安装CUDA(WSL2中操作)

参考CUDA官网,选择自己对应的版本。

这里的cuda版本,要比刚才cmd中标注的CUDA version要低。

比如我的CUDA version是11.6,我在下载CUDA驱动时,可以安装最新的11.5版本。

一定要选择红圈标注的WSL版本!!!

选择cuda版本及安装模式

选择完版本后,下面会有整个安装提示,在WSL2中按照方法来就行,如:

在WSL中一步一步地跟着安装即可

安装相关依赖库

在Ubuntu上配置CUDA的环境变量,命令行输入:

在末尾添加(注意:将地址里面的11.5换成自己安装的版本!):

更新文件:

输入代码检测是否安装完成

如上图,即安装成功

4.安装CUDNN(在WSL2中安装)

需要注册账号,参考如下地址的3.2

然后可以安装pytorch,可以直接安装最新版本,只要比驱动version低就行。



【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3