TensorFlow入門教程1-安裝TensorFlow

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版本,下載,

圖片.png

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之旅。
————————————————

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容