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

Z.Q.Feng Z.Q.Feng     2023-01-24     157

关键词:

文章目录


一、前言

Ubuntu 下安装 TensorFlow 并不是简单的使用 pip 安装就可以的,如果不能使用 GPU 的话效率是会很低的,这里的方法都是自己在系统下搭建踩过一些坑摸索出来的,希望会有些帮助。

  • 系统:Ubuntu21.10
  • cuda版本:cuda11.1
  • cudnn版本:cudnn8.1.0

二、安装NVIDIAdriver

1. 添加PPA源

添加 PPA 源:

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

为系统安装依赖项以构建内核模块:

sudo apt-get install dkms build-essential

2. 查看显卡硬件型号

打开终端,输入如下命令

ubuntu-drivers devices

输出界面如下,可以看到这里我的硬件显卡型号为 GP108M [GeForce MX250],推荐安装的驱动版本号为:driver : nvidia-driver-470 - distro non-free recommended

3. 安装NVIDIA驱动

这里我们下载自己电脑推荐的驱动版本即可:

sudo ubuntu-drivers autoinstall

系统会自动安装推荐版本驱动,接着重启系统。


三、安装cuda

这里我用的 cuda 版本为 11.1,采用 .run 文件安装,最新的 11.6 版本试了有点问题,所以我换成了 11.1 版本。

1. 下载相关依赖

命令如下:

sudo apt-get install freeglut3 freeglut3-dev libxi-dev libxmu-dev

2. 下载安装文件

直接使用 wget 下载即可,这里我提供的是 11.1 版本的下载链接:

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

大概有 3 个多 G,注意网络连接。

3. 安装cuda-toolkit

这里我们只使用安装文件来安装 cuda-toolkit,driver 的安装我们在上述操作中已经完成,所以命令如下:

sudo sh ./cuda_11.1.0*.run --toolkit --silent --override

4. 配置环境变量

修改 ~/.bashrc 文件:

vim ~/.bashrc

添加如下内容:

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

使修改立即生效:

source ~/.bashrc

5. 检验安装

命令如下:

nvcc --version


四、安装cudnn

1. 下载安装文件

这里我下载的是 8.1.0 版本,cudnn 的安装需要与你的 cuda 版本对应,官网下载链接如下:https://developer.nvidia.com/rdp/cudnn-archive,你可能需要注册才能下载(国内邮箱即可注册)。

选择 cuDNN Library for Linux (x86_64) 下载。

2. 安装cudnn8.1.0

进入下载目录:

cd ~/Downloads

解压 cudnn-11.2-linux-x64-v8.1.0.77.tgz 包:

mkdir cudnn-11.1
tar -zxf cudnn-11.2-linux-x64-v8.1.0.77.tgz -C ./cudnn

进入解压后文件目录:

cd cudnn

安装执行文件并赋予权限:

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

查看 cudnn 版本:

cat /usr/local/cuda/include/cudnn_version.h

3. 配置环境变量

编辑 ~/.bashrc 文件:

vim ~/.bashrc

添加内容如下:

export LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH

source 一下:

source ~/.bashrc

五、安装TensorFlow

1. 安装依赖包

安装 TensorFlow 之前需要我们安装两个个依赖包,这里我的 cuda 版本为 11.1,cudnn 版本为 8.1.0,下载依赖包为

  • libcudnn8_8.1.0.77-1+cuda11.2_amd64.deb
  • libcudnn8-dev_8.1.0.77-1+cuda11.2_amd64.deb

官网链接如下:https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/

这里我使用 wget 下载:

cd ~/Downloads
wget https://developer.download.nvidia.cn/compute/cuda/repos/ubuntu2004/x86_64/libcudnn8_8.1.0.77-1+cuda11.2_amd64.deb
wget https://developer.download.nvidia.cn/compute/cuda/repos/ubuntu2004/x86_64/libcudnn8-dev_8.1.0.77-1+cuda11.2_amd64.deb

安装命令如下:

sudo dpkg -i libcudnn8_8.1.0.77-1+cuda11.2_amd64.deb
sudo dpkg -i libcudnn8-dev_8.1.0.77-1+cuda11.2_amd64.deb

2. 安装TensorFlow

直接用 pip 安装即可,这里要求 Python 版本为 3.6-3.9pip >= 19.0,先使用如下命令升级你的 pip 工具:

python3 -m pip install --upgrade pip

使用 pip 安装 TensorFlow,这里我选择的是阿里云:

pip install -i https://mirrors.aliyun.com/pypi/simple tensorflow

3. 测试

打开 Python,使用如下命令测试:

import tensorflow as tf
print(tf.__version__)

输出如下:

接着测试一下我们的 GPU 支持:

tf.config.list_physical_devices("GPU")

大功告成!


六、总结

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

...3;CUDA8.0深度学习主机环境配置:Ubuntu16.04+GeForceGTX1080+TensorFlow不过在实际运行的过程中,有一定的不同之处,随着时间的推移,一些组件已经可以更方便的安装, 查看详情

ubuntu16安装tensorflow及jupyternotebook(代码片段)

Ubuntu16安装TensorFlow及Jupyternotebook文章目录Ubuntu16安装TensorFlow及Jupyternotebook1.软件版本2.安装过程3.测试本文主要讲述Windows上VMware虚拟机中Ubuntu16.04(最小化安装版,没有安装界面环境)安装TensorFlow及Jupyternotebook的过程1.... 查看详情

ubuntu16安装tensorflow及jupyternotebook(代码片段)

Ubuntu16安装TensorFlow及Jupyternotebook文章目录Ubuntu16安装TensorFlow及Jupyternotebook1.软件版本2.安装过程3.测试本文主要讲述Windows上VMware虚拟机中Ubuntu16.04(最小化安装版,没有安装界面环境)安装TensorFlow及Jupyternotebook的过程1.... 查看详情

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

...3;CUDA8.0深度学习主机环境配置:Ubuntu16.04+GeForceGTX1080+TensorFlow不过在实际运行的过程中,有一定的不同之处,随着时间的推移,一些组件已经可以更方便的安装,不再需要自己编译了。一些流程也有所更改。因... 查看详情

ubuntu21.10下安装使用armadillo库(代码片段)

文章目录一、前言二、下载安装文件三、编译与安装四、代码示例五、总结一、前言Armdillo矩阵运算速度跟MATLAB一个量级,为目前使用比较广的C++矩阵运算库之一,是在C++下使用MATLAB方式操作矩阵很好的选择&#... 查看详情

eiqwsl下工具及环境配置

...7.html2.配置VScode参考codehttps://zhuanlan.zhihu.com/p/812871912.配置tensorflowr1.12cd/mnt/c/yourdirgitclone[https://github.com/tensorflow/tensorflow.git]或者直接下载zip文件安装Bazel安装依赖环境$sudoapt-getinstallbuild-essentialopenjdk-11-jdkpythonzipunzip下载源码[https://g... 查看详情

ubuntu下自动安装tensorflow脚本

(1)前提是你得安装好驱动和cuda安装驱动和cuda很简单,到以下位置选择ubuntu的cuda包即可https://developer.nvidia.com/cuda-toolkit然后sudodpkg-i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.debsudoapt-getupdatesud 查看详情

ubuntu环境下anaconda安装tensorflow并配置jupyter远程访问(代码片段)

... 本文主要讲解在Ubuntu系统中,如何在Anaconda下安装TensorFlow以及配置JupyterNotebook远程访问的过程。  在官方文档中提到,TensorFlow的安装主要有以下五种形式:Pip安装:这种安装形式类似于安装其他的Python安装包... 查看详情

ubuntu16.04下安装tensorflow(gpu)

参考:http://blog.sina.com.cn/s/blog_672f698e0102wavp.html1.首先安装nvidia显卡驱动:系统设置->软件更新->附加驱动->选择nvidia最新驱动(361)->应用更改?2.下载CUDA8.0地址https://developer.nvidia.com/cuda-release-candidate-download( 查看详情

ubuntu16.04下安装tensorflow(gpu加速)----详细图文教程

...no,但是发现Theano天书般的源码真是头大,在看到tensorflow中文教程后,发现它竟然逻辑清晰,教程丰富,实在是居家旅行必备良药啊![偷笑][偷笑][偷笑]所以决定利用国庆假期学习ubuntu和TensorFlow的安装&# 查看详情

求助tensorflow下遇到cudacomputecapability问题

...装教程可以参考:Ubuntu16.04上gtx1080的cuda安装July172016目前tensorflow是一个非常流行的深度学习计算框架,常规硬件及系统的安装方法官方的doc已经说的很清楚了,但是因为系统是ubuntu16.04,显卡是GTX1080,所以不可避免的要折腾起来。... 查看详情

snap版本的firefox将在ubuntu21.10成为默认

近日,由Canonical的OlivierTilloy提交的FFE显示,Ubuntu将用.snap版本的Firefox替换Ubuntuseed中的Firefox.deb包。他在提交中写道:"根据Canonical与Mozilla的发行协议,从Ubuntu21.10开始,我们将在桌面ISO上默认安装.snap版本的... 查看详情

anaconda3tensorflow安装及modulenotfounderror:nomodulenamed'tensorflow'解答

Anaconda3的安装,参考:手把手教你如何安装Tensorflow(Windows和Linux两种版本)tensorflow的安装,参考:深度学习(TensorFlow)环境搭建:(三)Ubuntu16.04+CUDA8.0+cuDNN7+Anaconda4.4+Python3.6+TensorFlow1.3我选的是CPU下的安装,参考以上链接安装... 查看详情

使用tensorflow及anaconda(spyder)时遇到的问题(代码片段)

(1)问题一:如何在tensorflow环境下使用spyder答:在anacondanavigator中environment中搜索tensorflow,安装适合tensorflow的spyder(2)问题二:在在tensorflow环境下使用spyder时有些库文件(比如matplotlib)显示nomodule,如何解决答:anaconda下已经... 查看详情

centos6安装tensorflow及tensorflowonspark(代码片段)

....需求描述在Centos6系统上安装Hadoop、Spark集群,并使用TensorFlowOnSpark的YARN运行模式下执行TensorFlow的代码。(最好可以在不联网的集群中进行配置并运行)2.系统环境(拓扑)操作系统:Centos6.5Final;Hadoop&... 查看详情

ubuntu14.04安装tensorflow始末(代码片段)

基于ubuntu14.04干净的系统一步步遇到的坑记录下来;怀着平静学习的心情,问题总的能解决的! 1.首先看了下当前python版本python--versionPython2.7.6 2.安装pip  先更新源 sudoapt-getupdate sudoapt-getupgrade  安装pip  sudoapt-get... 查看详情

(tensorflow安装)查看tensorflow是否安装成功

...、打开命令行,输入pipshowtenwsorflow 命令,查看是否有tensorflow的相关版本、存储位置等信息。(如果报错,请参考:问题解决1)2、之后进入python环境输入importtenwsorflow如果没有报错,证明安装成功。(如果报错,请参考:问... 查看详情

tensorflow学习资源

 [1]安装Tensorflow(Linuxubuntu) http://blog.csdn.net/lenbow/article/details/51203526 [2]ubuntu下CUDA编译的GCC降级安装 http://blog.csdn.net/lenbow/article/details/51596706 [3]ubuntu手动安装最 查看详情