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版本,下载,
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之旅。
————————————————