ubuntu16.04下安装cuda,cudnn及tensorflow-gpu版本过程(代码片段)

multiangle multiangle     2022-12-07     532

关键词:


这篇文章主要依据两篇文章:
深度学习主机环境配置: Ubuntu16.04+Nvidia GTX 1080+CUDA8.0
深度学习主机环境配置: Ubuntu16.04+GeForce GTX 1080+TensorFlow
不过在实际运行的过程中,有一定的不同之处,随着时间的推移,一些组件已经可以更方便的安装,不再需要自己编译了。一些流程也有所更改。因此我在这里把自己在ubuntu16.04下安装cuda,cudnn和gpu版tensorflow的流程写下来,供人参考。


安装显卡驱动

首先安装显卡驱动。首先看自己显卡

lspci | grep -i vga
lspci | grep -i nvidia

然后看显卡驱动

lsmod | grep -i nvidia

在ubuntu16.04中,更换驱动非常方便,去
系统设置->软件更新->附加驱动->切换到最新的NVIDIA驱动即可。应用更改->重启

再运行nvidia-smi来看看


安装CUDA

https://developer.nvidia.com/cuda-downloads,根据自己的系统下载对应安装包,下载的是runfile文件,而不是deb文件。下载好以后,运行

sudo ./cuda_8.0.44_linux.run --tmpdir=/tmp

开始安装。后面参数 tmpdir好像是为了放日志,具体是什么日志忘了=。= 安装开始以后,首先是一个协议,一直按回车到底以后,输入accept。其他的操作如下所示

Do you accept the previously read EULA?
accept/decline/quit: accept

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 367.48?
(y)es/(n)o/(q)uit: n

Install the CUDA 8.0 Toolkit?
(y)es/(n)o/(q)uit: y

Enter Toolkit Location
[ default is /usr/local/cuda-8.0 ]:

Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y

Install the CUDA 8.0 Samples?
(y)es/(n)o/(q)uit: y

下面是安装时的输出信息:

Installing the CUDA Toolkit in /usr/local/cuda-8.0 …
Installing the CUDA Samples in /home/textminer …
Copying samples to /home/textminer/NVIDIA_CUDA-8.0_Samples now…
Finished copying samples.

===========
= Summary =
===========

Driver: Not Selected
Toolkit: Installed in /usr/local/cuda-8.0
Samples: Installed in /home/textminer

Please make sure that
– PATH includes /usr/local/cuda-8.0/bin
– LD_LIBRARY_PATH includes /usr/local/cuda-8.0/lib64, or, add /usr/local/cuda-8.0/lib64 to /etc/ld.so.conf and run ldconfig as root

To uninstall the CUDA Toolkit, run the uninstall script in /usr/local/cuda-8.0/bin

Please see CUDA_Installation_Guide_Linux.pdf in /usr/local/cuda-8.0/doc/pdf for detailed information on setting up CUDA.

***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 361.00 is required for CUDA 8.0 functionality to work.
To install the driver using this installer, run the following command, replacing with the name of this run file:
sudo .run -silent -driver

Logfile is /tmp/cuda_install_6583.log

安装完毕后,再声明一下环境变量,并将其写入到 ~/.bashrc 的尾部:

export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64$LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH

保存退出,运行source ~/.bashrc
测试是否安装成功

// 如果怕把samples搞坏了那就先搞一个备份,在备份里搞
cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery

结果如下

之前有看到说要把gcc改成4.9的。不过我现在用5.4的也可以编译,就是有几个警告,不知道会不会有什么大的影响。


cuDNN安装

如果要使用gpu来对tensorflow进行加速,除了安装CUDA以外,cuDNN也是必须要安装的。跟cuda一样,去nvidia的官网下载cuDNN的安装包。不过这次没法直接下载,需要先注册,然后还要做个调查问卷什么的,稍微有点麻烦。我下的是cuDNN v5.1 Library for Linux这个版本。不要下cuDNN v5.1 Developer Library for Ubuntu16.04 Power8 (Deb)这个版本,因为是给powe8处理器用的,不是amd64.

下载下来以后,发现是一个tgz的压缩包,使用tar进行解压

tar -xvf cudnn-8.0-linux-x64-v5.1.tgz

安装cuDNN比较简单,解压后把相应的文件拷贝到对应的CUDA目录下即可

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

tensorflow安装

我之前已经安装了cpu-only版的tensorflow,所以现在要先把原先的tf卸载

sudo pip uninstall tensorflow

之后从https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-0.12.0rc1-cp35-cp35m-linux_x86_64.whl下载gpu版的tf

sudo pip install tensorflow_gpu-0.12.0rc1-cp35-cp35m-linux_x86_64.whl

来测试一下

$ipython
import tensorflow as tf

结果如下

没有报错,说明已经安装成功了。


在idea中调用tensorflow

之前在命令行中已经调用tensorflow成功了,没想到在idea中写程序时调用tensorflow又出现了错误:

ImportError: libcudart.so.8.0: cannot open shared object file: No such file or directory

Error importing tensorflow.  Unless you are using bazel,
you should not try to import tensorflow from its source directory;
please exit the tensorflow source tree, and relaunch your python interpreter
from there.

但是我并没有在tf的根目录下运行呀?而且同一个文件用命令行是可以运行的呀?为什么换idea来就不行了捏?
后来经过尝试发现,问题还是出在环境变量LD_LIBRARY_PATH中。光把环境变量写在~/.bashrc中是不行的,还需要写在/etc/profile下。因此,可以在/etc/profile的结尾处加上

## cuda
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64$LD_LIBRARY_PATH:+:$LD_LIBRARY_PATH

登出后重新登陆,再打开idea,就可以在idea中调用tf啦

linux系统知识汇总(代码片段)

1系统相关1.1静态IP地址配置Ubuntu配置和修改IP地址1.2Linux内核升级和降级内核升级Linux升级内核的正确姿势内核降级Ubuntu16.04内核降级1.3清理缓存cacheubuntu手动释放缓存(清理内存cache)2JupyterLab2.1远程服务器访问Ubuntu服务器JupyterNoteboo... 查看详情

求助tensorflow下遇到cudacomputecapability问题

...然后计算能力高于3.0,具体可以查:安装教程可以参考:Ubuntu16.04上gtx1080的cuda安装July172016目前tensorflow是一个非常流行的深度学习计算框架,常规硬件及系统的安装方法官方的doc已经说的很清楚了,但是因为系统是ubuntu16.04,显卡... 查看详情

ubuntu21.10下安装tensorflow及配置gpu支持(cuda11.1+cudnn8.1.0)(代码片段)

...nsorFlow1.安装依赖包2.安装TensorFlow3.测试六、总结一、前言Ubuntu 查看详情

ubuntu16.04上安装cuda10.1和cudnnv8.0.5

参考技术A  直接运行cuda_10.1.105_418.39_linux.run,然后按照要求一步步操作就可以了。这里需要注意的是,因为已经安装了其他版本的cuda和驱动。所以,在安装的时候,不需要安装驱动。  其实,这里最主要想说的... 查看详情

ubuntu16.04+titanxp安装显卡驱动+cuda9.0+cudnn+其他软件(代码片段)

硬件环境ubuntu16.04LTS+windows10双系统NVIDIATiTanXP显卡(12G)软件环境搜狗输入法显卡驱动:LINUXX64(AMD64/EM64T)DISPLAYDRIVER(418.56)  https://www.nvidia.cn/Download/index.aspx?CUDA:Cuda9.0  https://develop 查看详情

普通电脑pc怎样跑tensorflow的gpu模式

...然后计算能力高于3.0,具体可以查:安装教程可以参考:Ubuntu16.04上gtx1080的cuda安装July172016目前tensorflow是一个非常流行的深度学习计算框架,常规硬件及系统的安装方法官方的doc已经说的很清楚了,但是因为系统是ubuntu16.04,显卡... 查看详情

tensorflowgpu版本运行时怎么知道有没有调用gpu以及cuda加速

...然后计算能力高于3.0,具体可以查:安装教程可以参考:Ubuntu16.04上gtx1080的cuda安装July172016目前tensorflow是一个非常流行的深度学习计算框架,常规硬件及系统的安装方法官方的doc已经说的很清楚了,但是因为系统是ubuntu16.04,显卡... 查看详情

ubuntu16.04无法安装cuda吗

参考技术AUbuntu16.04安装CUDA7.5作者:autocyz在介绍Ubuntu16.04安装CUDA7.5开始前,先辨析几个概念GPU、NVIDIA、NVIDIA驱动、CUDA、cudnn等,这些概念对于一个新手来说肯定是很晕的,正如我当初一样,所以我这里就稍微介绍一下这几个概念... 查看详情

ubuntu16.04安装nvidia驱动cuda,cudnn和tensorflow-gpu(代码片段)

本文个人博客地址:点击查看之前有在阿里云GPU服务器上弄过:点击查看,这里从装Nvidia开始一、安装Nvidia驱动1.1查找需要安装的Nvidia版本1.1.1官网官网上查找:https://www.nvidia.com/Download/index.aspx?lang=en-us这里是Ge... 查看详情

[cuda][cudnn][小白向]ubuntu环境下指定版本安装教程

...持的相关驱动程序(使用runfile会随cuda一起安装)ubuntu已经切换到了独显模式ubuntu中已经将之前安装的CUDA及cudnn卸载干净卸载cuda的方法sudoapt-getremovecudasudoaptautoremovesudoapt-getremovecuda*#删除对应的cuda版本文件夹cd/usr/local/sudor... 查看详情

ubuntu下安装cuda9.0和cudnn7.1(代码片段)

...:1.Pre-installationActions这一步需要检查GPU是否支持CUDAUbuntu版本是否受支持gcc版本检查KernelHeadersandDevelopmentPackages是否已安装这些步骤官方教程很详细,请移步官方教程第二节Pre-installationActions2.下载NividiaCudaT 查看详情

ubuntu16.04安装nvidia驱动cuda,cudnn和tensorflow-gpu(代码片段)

...ff0c;推荐410版本的1.1.2命令行查看推荐驱动查看驱动:ubuntu-driversdevices,如下图ubuntu@ubuntu-System-Product-Name:~$ubuntu-driversdevices==cpu-microcode.py==driver:intel-microcode-distrofree==/sys/devices/pci0000:00/0000:00:01.0/0000:01:00... 查看详情

cudnn安装及查看版本(代码片段)

?解压下载的tgz文件,解压后是一个cuda文件夹?使用命令进入cuda文件夹的位置?输入命令:sudocpcuda/include/cudnn.h/usr/local/cuda/include/sudocpcuda/lib64/libcudnn*/usr/local/cuda/lib64/-dsudochmoda+r/usr/local/cuda/include/cudnn.hsudochmoda+r/usr/local/cuda/lib64/libcudn... 查看详情

ubuntu16.04下安装cuda8.0

一、首先安装NVIDIA显卡驱动通过NVIDIA-Linux-x86_64-367.44.run文件安装。1.添加PPA.sudoadd-apt-repositoryppa:graphics-drivers/ppa2.刷新并安装新的驱动程序sudoapt-getupdatesudoapt-getinstallnvidia-367nvidia-settings(如果需要的话,)卸载:删除所有的nvi 查看详情

从头开始安装ubuntu,cuda,cudnn,caffe,tensorflow,ros

前言安装ubuntu一直是一件特别特别麻烦的事情,以前常常因为换一台主机就得完全重新安装一遍,每次安装都要折腾特别久,总会遇到各种各样的问题。因此很有必要写一个Blog完整记录整个过程及遇到的问题。现在&#... 查看详情

ubuntu16.04+cuda8.0+gtx1080安装教程

1.安装Ubuntu16.04不考虑双系统,直接安装Ubuntu16.04,从 ubuntu官方 下载64位版本:ubuntu-16.04-desktop-amd64.iso。在MAC下制作了UbuntuUSB安装盘,具体方法可参考: 在MAC下使用ISO制作Linux的安装USB盘 ,之后通过Bios引导U盘启动安... 查看详情

ubuntu18.04安装nvidiadriver+cuda+cudnn

LinkUbuntu18.04安装Anaconda3 查看详情

ubuntu18.04安装nvidiadriver+cuda+cudnn

LinkUbuntu18.04安装Anaconda3 查看详情