ubuntu18.04安装nvidia_driver_510+cuda_11.6+cudnn_11.x

一、安装nvidia_driver

ubuntu18.04安装nvidia_driver_510+cuda_11.6+cudnn_11.x

2.1首先我们需要添加源

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

2.2选择一个版本安装即可(如1,我选择安装510)

sudo apt install nvidia-driver-510

2.3 重启电脑后终端输入

nvidia-smi

查看驱动信息

ubuntu18.04安装nvidia_driver_510+cuda_11.6+cudnn_11.x

二、安装CUDA

1、官网找到自己的版本(如2.3版本信息中,我510驱动对应的cuda_11.6)

CUDA Toolkit Archive | NVIDIA Developer 选择runfile格式的CUDA文件下载

ubuntu18.04安装nvidia_driver_510+cuda_11.6+cudnn_11.x

3、下载完成后,解压,并运行上图中的命令,会有条款,接受即可。

3.1注意安装CUDA的时候不要安装驱动(因为在第一步我们已经安装过了)

ubuntu18.04安装nvidia_driver_510+cuda_11.6+cudnn_11.x
sudo gedit ~/.bashrc

在打开的txt文件末尾加

export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-11.6/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存,退出。终端执行

source ~/.bashrc

3.3 验证cuda

3.3.1

11.6版本cuda的安装目录/usr/local/cuda-11.6/samples里只有一个txt文件,大致意思是告诉你新版本的cuda,samples中内容需要自己在github下载。

由于github下载过慢,在此放上gitee链接

git clone https://gitee.com/liwuhaoooo/cuda-samples.git

在samples文件夹下打开终端执行上述语句。

大概率无权访问,此时在cuda-11.6文件夹下打开终端

su
输入密码切换超级用户
chmod 777 samples

再次执行git clone 就可以了。

3.3.2

进入/usr/local/cuda-11.6/samples/cuda-samples/Samples

cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

输出

ubuntu18.04安装nvidia_driver_510+cuda_11.6+cudnn_11.x

则安装成功。

三、安装cuDNN

1、

进入NVIDIA cuDNN | NVIDIA Developer注册,并选择合适的版本下载(cuDNN Library for Linux),然后解压;

2、

并进入到/home/lwh/Downloads/cudnn-11.3-linux-x64-v8.2.1.32目录,运行以下命令:

sudo cp cuda/include/cudnn.h /usr/local/cuda-11.6/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.6/lib64
sudo chmod a+r /usr/local/cuda-11.6/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.6/lib64/libcudnn*

若无权访问,像3.3.1一样,分别更改include和lib64文件夹权限。

四、验证GPU可用

import torch
print(torch.cuda.is_available())

ubuntu18.04安装nvidia_driver_510+cuda_11.6+cudnn_11.x

Original: https://blog.csdn.net/weixin_54424184/article/details/122654844
Author: CPU疼
Title: ubuntu18.04安装nvidia_driver_510+cuda_11.6+cudnn_11.x

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/496126/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球