从事深度学习无论是tensorflow还是caffe都需要安装cuda和cudnn这2个显卡支持的库。cuda安装容易失败,为了避免安装失败,请大家一定按照官方指导文件流程进行操作。下面也给出我的一些理解,给大家讲解一些该注意的地方,增加大家的理解。
一、支持框架验证
必须强调一下,一定要检查自己的系统版本、显卡、gcc、Linux内核是否支持cuda10.1。下图是支持cuda10.1的系统要求。使用以下指令查询自己的显卡、gcc、Linux内核。
$ lspci | grep -i nvidia
$ uname -m && cat / etc / * release
$ gcc --version
$ uname -r
二、安装
(1)执行上述框架验证,完成相应操作
(2)下载cuda toolkit,在下面的网址中进行下载:
https://developer.nvidia.com/cuda-10.1-download-archive-base
按照自己的版本选择,下面是我根据自己的电脑系统选择的,其中installer type(安装类型)runfile是整个安装包包含了所有的库,适用本地系统。deb是另一种安装形式,也适合本地系统。后面的,作为学习者,