关键词:
文章目录
一、前言
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.9
, pip >= 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手动安装最 查看详情