优秀的编程知识分享平台

网站首页 > 技术文章 正文

NVIDIA CUDA & cuDNN driver安装

nanyue 2024-08-08 19:09:12 技术文章 11 ℃

一、实验环境

CentOS 7

cuda_9.1.85_387.26_linux

cudnn-9.1-linux-x64-v7

NVIDIA_GeForce GTX 750(别问GPU为啥这么low,还不是因为qiong)

二、安装CUDA

2.1下载CUDA:

cuda_9.1.85_387.26_linux(https://developer.nvidia.com/cuda-downloads)

chmod +x cuda_9.1.85_387.26_linux

2.2关闭nouveau:

/etc/modprobe.d/nvidia-installer-disable-nouveau.conf:

blacklist nouveau

options nouveau modeset=0

mv /boot/initramfs−sudomv/boot/initramfs−(uname -r).img /boot/initramfs-(uname -r).img.bak

dracut /boot/initramfs-(uname−r).img

2.3 下载kernel-devel&kernel-headers

yum install kernel-devel

yum install kernel-headers

2.4 reboot

2.5 ./cuda_9.1.85_387.26_linux

2.6 nvidia-smi

三、安装cuDNN

3.1 下载cuDNN v7.0.5 Library for Linux:

3.2 解压cuDNN v7.0.5 Library for Linux:

cuDNN安装到/usr/local/cuda(即:/usr/local/cuda-9.1)

最近发表
标签列表