Windows安裝tensorflow教程 GPU版


 

前置准備

查看GPU型號

電腦桌面->右鍵我的電腦->選擇管理->點擊設備管理器  如下圖:

 

如果不是英偉達顯卡,那么不用往下看了,GAMEOVER!

 

查看CUDA算力

gpu版本要求電腦的GPU硬件必須有CUDA支持,並且計算能力最低為3.5以上。

查看地址在這里:https://developer.nvidia.com/cuda-gpus

這個就是我的:

 

下載GPU驅動

下載地址:https://www.nvidia.com/download/index.aspx?lang=en-us#

我的演示:

 

這個驅動的版本號必須要達到418.x以上

 

下載Anaconda

清華大學鏡像地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/

我下載的版本是:Anaconda3-5.3.1-Windows-x86_64.exe

配置高級選項的時候  第一個我選擇將Anaconda加入環境變量  第二個注冊Python3.5沒有勾選,因為我電腦上安裝得有python3.7。

 

配置Anaconda軟件包下載服務器

從開始菜單中打開Anaconda Prompt,依次輸入以下三條命令

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

 

新建虛擬環境

Anaconda默認的虛擬環境是base(root),其中已經安裝的軟件包太多,並且這個虛擬環境無法刪除,所以我們新建一個虛擬環境進行學習,將來不想使用的時候可以將新的虛擬環境刪除。

依然在Anaconda Prompt的命令行中,執行命令:

conda create -n tf_gpu python=3.7

其中conda create是創建命令  -n是name的意思  后面tf_gpu是新環境名 名字隨意取  python=3.7是設置該環境的python版本 一般來說3.6和3.7兩種

 

 

這里輸入y  表示同意安裝相關軟件包

 

現在新的虛擬環境已經創建完成

 

執行命令:

conda activate tf_gpu

激活tf_gpu這個環境,也可以理解成進入tf_gpu環境。

然后你會看到頭部的(base)變成了(tf_gpu)

 

安裝TensorFlow

接着上面的命令行窗口

安裝英偉達SDK

conda install cudatoolkit=10.1

 

輸入y  同意安裝相關軟件包

 

安裝英偉達深度學習軟件包

conda install cudnn=7.6

 

只要不是下面這個樣子,一般都是安裝失敗,仔細看是否有error字樣,如果安裝失敗,多試幾次,我就失敗了兩次,第三次才成功,可能是網絡抖動而導致的。

 

安裝TensorFlow

pip install tensorflow-gpu==2.1.2

一連串的下載安裝信息滑過命令行之后,就算安裝完了。

 

校驗是否安裝成功

繼續命令行輸入python 進入解釋器

輸入兩行:

import tensorflow as tf
tf.__version__

如果打印出版本號'2.1.2' 則安裝成功。

 

打開pycharm

新建項目 選擇conda環境解釋器  如圖:

 

至此,環境已經搭建完成,可以進行項目編程了。

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM