Linux安装Nvidia显卡驱动+CUDA+cuDNN+PyTorch

本文详细介绍了在Ubuntu18.04环境下安装Nvidia、CUDA、cuDNN和PyTorch显卡驱动的整个过程。
下面是具体步骤和内容:首先,检查您电脑的显卡型号。
本例中显卡型号为GeForceGTX1080Ti。
接下来,禁用Ubuntu自带的Nouveau驱动,以确保使用Nvidia官方驱动。
这一步对于避免兼容性问题非常重要。
编辑该文件以禁用Nouveau驱动程序。
添加特定代码以启用禁用。
更新系统并重新启动计算机以使更改生效。
检查Nouveau驱动程序是否已成功禁用。
如果没有输出,则说明已禁用成功。
安装GCC和CMake。
检查安装状态,确认安装成功。
下载Nvidia显卡驱动,进入官网,选择对应的显卡型号,下载指定版本。
关闭显示模式,切换到tty终端模式并输入命令关闭显示管理器。
在终端中运行下载的驱动程序安装程序,然后按照说明继续安装。
重启后输入nvidia-smi查看驱动安装结果应该会显示CUDA的兼容版本。
安装CUDA,检查相关CUDA版本,下载CUDA10.0,安装并确认安装不完整提示。
配置环境变量以确保安装成功。
测试CUDA配置,输出类似“PASS”表示成功。
下载cuDNN,选择与CUDA版本对应的版本,下载并解压文件。
将cudnn.h和libcudnn*文件复制到CUDA安装对应目录的cuDNN文件夹中。
至此,cuDNN安装完成,可以开始安装PyTorch了。
访问PyTorch官网查找CUDA版本对应的PyTorch版本建议使用豆瓣源通过pip安装,以确保安装成功。
查看PyTorch是否成功调用显卡,并通过特定命令进行验证。
完成上述步骤后,您将在Ubuntu18.04环境中成功安装Nvidia显卡驱动、CUDA、cuDNN和PyTorch,并为使用GPU加速的深度学习项目做好准备。

如何安装nvidiageforceforLinux显卡驱程

Linux上安装nvidia显卡驱动:1.检查显卡是否安装:(1)运行glxinfo|greprendering;来自英伟达网站;:(1)开始安装前,更改第一个/etc/inittab文件,将:id:5:initdefault:改为:id:3:initdefault:#vi/etc/inittab(2)然后重新启动计算机,运行NVIDIA系统程序进入行为模式并使用root登录,然后运行nvidia-linux-x86-1.0-5336-pkg1.run#shNVIDIA-linux-x86-1.0-5336-pkg1.run(3)安装过程中,根据建议进行选择有旧驱动,询问是否删除旧驱动如果建议删除某个模块(s)缺失,系统询问您是否在线映射,选择否;选择它是;接下来等待安装完成(安装大约需要5分钟)。
(4)更改xorg.config文件,手动将Driver"nv"更改为Driver"nvidia"#vi/etc/X11/xorg.conf(5)最后更改inittab文件,将:id:3:initdefault:更改为:id:5:initdefault:安装完成,重新启动计算机#reboot(6)当看到NV标志时,表示安装完成。
(7)如果不想显示NVIDIALOGO,可以将/etc/X11/xorg.conf中“Device”部分的Option“NoLogo”“True”更改为:Section“Device”Identifier“UniversalDisplayCard”Driver"nvidia"Option"NoLogo""True"EndSection完成上述步骤后,nvidia驱动程序已在Linux上成功安装。