TensorFlow入门教程1-安装TensorFlow

1、概述

最近接触感知算法方向的研究,之前在CSDN上分享了自己的一些TensorFlow学习笔记,也得到了一些网友的认可,但是我感觉有点乱,有些地方写的也不够详细,所以想重新整理一下,希望对需要的人有所帮助。

2、电脑配置

首先说一下我的电脑配置:

操作系统:Ubuntu 20.4

3、安装python3.x(略)

4、安装CUDA

因为我用的是GPU版的TensorFlow,如果你没有GPU或者你比较任性,老子就是不用GPU,你就可以跳过这步。首先打开CUDA下载页面,

https://developer.nvidia.com/cuda-toolkit-archive

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.5.1/local_installers/cuda-repo-ubuntu2004-11-5-local_11.5.1-495.29.05-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-5-local_11.5.1-495.29.05-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-5-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

下载,双击安装,全部选默认即可。

5、安装cuDNN

首先进入下载界面,

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

如果没有账号则要先注册,然后找到相应的CUDA版本,下载,

图片.png

6、安装TensorFlow

接下来就是安装TensorFlow了,打开Anaconda的终端,

在命令行窗口中输入python,回车,得到下面的界面,

输入下面代码,

import tensorflow as tf

回车,得到下面的界面,

可以看到此时并没有安装TensorFlow。输入exit(),回车,退出Python命令行界面。

再在终端输入,

pip install tensorflow-gpu==1.15.19

回车,

我故意输入一个不存在的TensorFlow版本号,这时他就会提示我们能安装哪些版本,就是红色字体部分。2.X版本跟1.X版本有很大的不同,因为之前的博客用的都是1.X的版本,所以,我这里先安装1.15.0版本,以后如果有需要换版本再说。输入以下命令进行安装,

pip install tensorflow-gpu==1.15.0

如果看到上面这样的信息就说明安装成功了。这时候,再输入python,回车,输入

import tensorflow as tf

得到下面的界面说明安装成功了。

如果没有GPU要安装CPU的版本,则执行,

pip install tensorflow==1.15.0

即可。

这一讲,我们就先讲TensorFlow的安装,因为我发现有些网友连安装都出问题。新手嘛,都是这么过来的,下一讲,我们就真正的开启TensorFlow之旅。
————————————————

你可能感兴趣的:(TensorFlow入门教程1-安装TensorFlow)