linux上安装NVIDIA显卡驱动以及深度学习需要的cudn、cudnn、pytorch

1.安装显卡驱动必要的依赖

yum -y install pciutils
yum install -y epel-release
yum install -y kernel-headers kernel-devel dkms

2.执行shell脚本禁用nouveau驱动

cat >>/lib/modprobe.d/dist-blacklist.conf<<eof
blacklist nouveau
options nouveau modeset=0
eof
mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
mv /boot/initramfs-$(uname -r).img.bak /home/initramfs-$(uname -r).img.bakdracut /boot/initramfs-$(uname -r).img $(uname -r)
systemctl set-default multi-user.target

3.安装驱动Official Advanced Driver Search | NVIDIA

chmod +x NVIDIA-Linux-x86_64-460.106.00.run
sh NVIDIA-Linux-x86_64-460.106.00.run

验证是否安装成功

4.安装cuda

查看驱动对应的cuda版本Release Notes :: CUDA Toolkit Documentation

CUDA 工具包	CUDA 次要版本兼容性所需的最低驱动程序版本*
CUDA 工具包	Linux x86_64 驱动版本	Windows x86_64 驱动程序版本
CUDA 11.6.x	>=450.80.02	>=452.39
CUDA 11.5.x	>=450.80.02	>=452.39
CUDA 11.4.x	>=450.80.02	>=452.39
CUDA 11.3.x	>=450.80.02	>=452.39
CUDA 11.2.x	>=450.80.02	>=452.39
CUDA 11.1 (11.1.0)	>=450.80.02	>=452.39
CUDA 11.0 (11.0.3)	>=450.36.06**	>=451.22**

wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda-repo-rhel7-11-3-local-11.3.0_465.19.01-1.x86_64.rpm
sudo rpm -i cuda-repo-rhel7-11-3-local-11.3.0_465.19.01-1.x86_64.rpm
sudo yum clean all
sudo yum -y install nvidia-driver-latest-dkms cuda
sudo yum -y install cuda-drivers

配置cuda环境变量

vi /etc/profile文件末尾添加
export CUDA_HOME=/usr/local/cuda-11.3
export PATH=/usr/local/cuda-11.3/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib:$LD_LIBRARY_PATH刷新环境变量使其生效source /etc/profile

验证是否安装成功

5.安装cudnn https://developer.nvidia.com/rdp/cudnn-archive

tar -zxvf cudnn-11.3-linux-x64-v8.2.1.32.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

6.安装conda

wget https://repo.anaconda.com/archive/Anaconda3-2021.11-Linux-x86_64.shchmod +x Anaconda3-2021.11-Linux-x86_64.sh./Anaconda3-2021.11-Linux-x86_64.shsource ~/.bashrc

验证是否安装成功

7.安装pytorch

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

查看是否安装成功

返回true就是安装成功了

linux上安装NVIDIA显卡驱动以及深度学习需要的cudn、cudnn、pytorch相关推荐

在Redhat Linux系统上安装NVIDIA显卡驱动的详细教程
教程目录一.下载安装NVIDIA显卡驱动 1.查看自己的显卡型号 2. 在官网下载对应的驱动: 3.禁用nouveau驱动 4.重塑开机镜像 5.重启计算机 6.检测nouveau是否禁用成功 7. ...
debian nvidia 安装_在 Debian 上安装 Nvidia 显卡驱动启用 Nvenc
最近弄了一台带 GT 710 显卡的杜甫,便想着可以利用 Nvenc 显卡硬件编码来驱动 Jellyfin 在线转码云播.不过折腾的过程中遇到了不少问题,在此梳理一番正确的安装流程,以便来日查询参考. ...
ubuntu 14.04 安装nvidia GTX745 驱动及深度学习环境搭建心得
在实验室装GPU driver发现出现各种各样的问题,分享一下我成功的步骤此时你可能已经装了Nvidia驱动,发现黑屏, ctrl + alt + F1 首先删除Nvidia所有驱动 sudo ap ...
Ubuntu14.04 安装 Nvidia显卡驱动
安装后遇到了循环登录的问题,这种情况下,重启进终端,将Nvidia驱动卸载,然后就可以正常登陆.一般这个问题,我的解决办法是,安装完成后,别更新系统内核就行! 机型为戴尔Vostro3900 显卡型号 ...
Linux 桌面玩家指南：11. 在同一个硬盘上安装多个 Linux 发行版以及为 Linux 安装 Nvidia 显卡驱动...
Linux 桌面玩家指南:11. 在同一个硬盘上安装多个 Linux 发行版以及为 Linux 安装 Nvidia 显卡驱动原文:Linux 桌面玩家指南:11. 在同一个硬盘上安装多个 Linux ...
linux装nvidia驱动没有图形界面,[转]Fedora 13下安装Nvidia显卡驱动的方法/Linux无法进入X Windows 图形界面的解决方法...
再试一试这个方法:[转自]http://www.myexception.cn/image/505352.html VirtualBox3.1.2安装Fedora12后无法启动图形界面问题 Virtua ...
在TITAN RTX 2080Ti 上安装 Ubuntu18.04+Nvidia-430显卡驱动+配置深度学习环境（1）
最近一个项目需要在TITAN RTX 2080Ti 上安装Ubuntu18.04+Nvidia-430显卡驱动+Cuda10.1+Cudnn+7.6+Anaconda3+深度学习环境(tensorfl ...
nvidia显卡linux 7.2,Centos7 安装 Nvidia 显卡驱动
系统环境1 2CentOS Linux release 7.6.1810 (Core) Linux centos7.6 3.10.0-957.5.1.el7.x86_64 #1 SMP Fri Feb ...
Linux安装NVIDIA显卡驱动并配置pytorch和tensorflow环境
一.系统环境:Ubuntu18.04+显卡3090 双系统安装:Win10安装Ubuntu18.04双系统二.安装NVIDIA显卡驱动: 1.进入BIOS禁用secure boot. 2.卸载原有的 ...
linux安装cuda需要先安装NVIDIA显卡驱动
CentOS7.3安装NVIDIA-1080ti驱动.cuda.cudnn.TensorFlow - 负赑屃 - 博客园使用这个教程学习语音之家付费教程踩坑实录在虚拟机安装NVIDIA显卡驱动失 ...

linux上安装NVIDIA显卡驱动以及深度学习需要的cudn、cudnn、pytorch

linux上安装NVIDIA显卡驱动以及深度学习需要的cudn、cudnn、pytorch相关推荐

最新文章

热门文章