安装 Ubuntu20.04安装 Cuda11和cuDNN

2020-10-4 23:34:49 来源: GNSSer 发布人:

安装 Ubuntu20.04安装 Cuda11cuDNN

CudaNvidia显卡计算环境,cuDNN是在Cuda基础上的深度学习库。

1   准备工作

1)      具有足够空间(10GB以上)的Ubuntu系统;

2)      支持Cuda的显卡,此处为 GTX 1650, 1000元左右。

3)      Internet网络连接,或提前下载。

2   安装Nvidia官方驱动

可选系统自动安装和手动下载安装。

https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 这里可以查询到各个版本支持的情况。

2.1 系统联网自动安装

Ubuntu的“软件和更新/附加驱动”中,选择使用NVIDIA driver,而不使用系统自带的nouveau

“应用更改”后系统下载驱动并安装。

2.2 手动安装

在以下网址下载Nvidia驱动,选择自己显卡的型号。

https://www.nvidia.cn/geforce/drivers/

需要转入完全的命令窗口中,卸载原来的驱动,然后再安装,最后加入路径。

整个过程较麻烦。请自行百度。

2.3 查看安装状态

查看驱动的安装状态

  1. $ sudo nvidia-smi
  2. $ sudo nvidia-settings


3   安装Cuda

3.1 下载cuda安装包

在如下网址选择合适的安装包

https://developer.nvidia.com/cuda-downloads

比如,本次配置为linuxx86_64ubuntu20.04runfile(local),得到如下安装提示。

文件较大(3GB),网速很快,10分钟不到就下载完了。

wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run

sudo sh cuda_11.1.0_455.23.05_linux.run

3.2 根据命令提示安装

默认设置即可。

注意:前面已经安装驱动了,因此这一步不需要安装驱动了,否则会报错。

可能的错误:

如果出现gcc报错,则需要安装gcc。命令如下:

sudo apt-get install gcc

 

4   安装cuDNN

4.1 下载

在如下网址下载cuDNN,当前需要注册和填写一个简答的问卷调查。

https://developer.nvidia.com/cudnn-download-survey

选择正确的版本,文件有好几个需要下载,有运行环境,开发环境和示例。

4.2 安装

下载的Deb文件,直接双击即可安装。

5   完成安装

根据提示,将路径加入系统PATH中。

添加路径方法.

 

6   测试安装

测试CUDAsammples,主要一本机为例,执行如下代码,有结果则表明安装正确。

cd /usr/local/cuda/samples/1_Utilities/deviceQuery #由自己电脑目录决定

make

sudo ./deviceQuery

 

 





阅读次数: 1050

下一篇: Ubuntu 安装 Anaconda
上一篇: Ubuntu下添加路径到PATH中的方法

尚无评论!

返回上一页面