1、概述
最近接觸感知算法方向的研究,之前在CSDN上分享了自己的一些TensorFlow學(xué)習(xí)筆記,也得到了一些網(wǎng)友的認(rèn)可,但是我感覺有點(diǎn)亂,有些地方寫的也不夠詳細(xì),所以想重新整理一下,希望對(duì)需要的人有所幫助。
2、電腦配置
首先說一下我的電腦配置:
操作系統(tǒng):Ubuntu 20.4
3、安裝python3.x(略)
4、安裝CUDA
因?yàn)槲矣玫氖荊PU版的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
下載,雙擊安裝,全部選默認(rèn)即可。
5、安裝cuDNN
首先進(jìn)入下載界面,
https://developer.nvidia.com/rdp/cudnn-download
如果沒有賬號(hào)則要先注冊,然后找到相應(yīng)的CUDA版本,下載,

6、安裝TensorFlow
接下來就是安裝TensorFlow了,打開Anaconda的終端,
在命令行窗口中輸入python,回車,得到下面的界面,
輸入下面代碼,
import tensorflow as tf
回車,得到下面的界面,
可以看到此時(shí)并沒有安裝TensorFlow。輸入exit(),回車,退出Python命令行界面。
再在終端輸入,
pip install tensorflow-gpu==1.15.19
回車,
我故意輸入一個(gè)不存在的TensorFlow版本號(hào),這時(shí)他就會(huì)提示我們能安裝哪些版本,就是紅色字體部分。2.X版本跟1.X版本有很大的不同,因?yàn)橹暗牟┛陀玫亩际?.X的版本,所以,我這里先安裝1.15.0版本,以后如果有需要換版本再說。輸入以下命令進(jìn)行安裝,
pip install tensorflow-gpu==1.15.0
如果看到上面這樣的信息就說明安裝成功了。這時(shí)候,再輸入python,回車,輸入
import tensorflow as tf
得到下面的界面說明安裝成功了。
如果沒有GPU要安裝CPU的版本,則執(zhí)行,
pip install tensorflow==1.15.0
即可。
這一講,我們就先講TensorFlow的安裝,因?yàn)槲野l(fā)現(xiàn)有些網(wǎng)友連安裝都出問題。新手嘛,都是這么過來的,下一講,我們就真正的開啟TensorFlow之旅。
————————————————