您现在的位置是:首页 >学无止境 >新Ubuntu怎么装Nidia驱动,cuda和cudnn网站首页学无止境

新Ubuntu怎么装Nidia驱动,cuda和cudnn

Jerry-hao 2024-09-06 12:01:03
简介新Ubuntu怎么装Nidia驱动,cuda和cudnn

怎么安装nvidia驱动

软件更新->附加驱动

选择一个喜欢的

或者找推荐的

Ubuntu 20.04安装CUDA & CUDNN 手把手带你撸_ubuntu20.04 无图形化安装cuda_哈希Map的博客-CSDN博客

sudo ubuntu-drivers autoinstall

怎么安装cuda

gcc 不用降级 (我没降级就安装好了)

1,去官网找合适的cuda并执行命令 wget 和sudo (下面有)

CUDA Toolkit Archive | NVIDIA Developer

2,安装cuda

 因为我们已经安装nvidia驱动了, 这里把驱动取消。

然后安装

3,添加环境变量

gedit ~/.bashrc

然后在打开文件的最后一行

添加:你的cuda是什么版本就改成什么!

export PATH="/usr/local/cuda-11.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH"

4,验证

输入 

nvcc -V

 成功

怎么安装cudnn

1,选择合适的版本下载(cuDNN Library for Linux)

cuDNN Archive | NVIDIA Developer

2,然后解压,并进入到相应目录,运行以下命令:

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


以配置cuDNN环境。

通过

cat /usr/local/cuda-11.1/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

可以查看cuDNN版本。

如果没有输出的话,

参考这个:关于 cat: /usr/local/cuda/include/cudnn_version.h: 没有那个文件或目录_cat没有那个文件或目录_qq_39658788的博客-CSDN博客

 把所有的文件都复制过去

sudo cp cuda/include/* /usr/local/cuda-11.1/include
sudo chmod a+r /usr/local/cuda-11.1/include/*

然后输入这个查看版本

cat /usr/local/cuda-11.1/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

 

说一下个人的猜想

/usr/local中有cuda 和cuda-11.1

那么我在想,cuda会不会类似于头文件,或者总的调用的库,可以用来和各个版本的cuda交互呢?

举个例子:我下载了cuda-11.1 和 cuda11.4那么,/usr/local中就有cuda,cuda-11.1,cuda-11.4

那么我下的cuda-11.1版本的cudnn的文件就复制到 cuda-11.1的里面,11.4的就复制到11.4的里面。

回到现实,我有cuda 和cuda-11.1,然后我下载了cudnn,我应该把cudnn的文件复制到cuda-11.1 中而不是 cuda

不知道这个观点对不对?

风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。