1. 程式人生 > 程式設計 >Window10上Tensorflow的安裝(CPU和GPU版本)

Window10上Tensorflow的安裝(CPU和GPU版本)

之前摸索tensorflow的時候安裝踩坑的時間非常久,主要是沒搞懂幾個東西的關係,就在瞎除錯,以及當時很多東西不懂,很多報錯也一知半解的。這次重灌系統後正好需要再配置一次,把再一次的經歷記錄一下。我的電腦是華為的matebook13,intel i5-8625U,MX250顯示卡,win10系統。(不得不吐槽很垃圾,只能滿足測試測試調調程式碼的需求)

深度學習利用Tensorflow平臺,其中的Keras Sequential API對新使用者非常的友好,可以將各基礎元件組合在一起來構建模型。

(官網: https://tensorflow.google.cn/?hl=zh-cn)

Window10上Tensorflow的安裝(CPU和GPU版本)

Window10上Tensorflow的安裝(CPU和GPU版本)

安裝Tensorflow 分為 tensorflow_cpu 和 tensorflow_gpu版本

GPU就是用來渲染計算的,GPU版本計算效能是CPU的百倍之快。如果電腦沒有獨立顯示卡只能用CPU版本計算。

Window10上Tensorflow的安裝(CPU和GPU版本)

CPU版本安裝:

tensorflow_cpu版本只需要安裝anaconda後在anaconda prompt裡面pip install tensorflow_cpu==(版本號) 即可。安裝anaconda的方法見GPU版本里面。

注意查下python和tensorflow_cpu適配的版本號。

Window10上Tensorflow的安裝(CPU和GPU版本)

GPU版本安裝

tensorflow_gpu版本安裝大致分為三步:1、安裝anaconda 2、安裝cuda和cdunn 3、安裝tensorflow_gpu。其中第一步和第二步的順序可以調換,就是安裝完了前面三個東西再安裝tensorflow即可。

tensorflow最近出了2.0版本,和1.0大版本有一些區別。具體我還沒有去了解,程式碼不一定相容,需要注意一下。這三個步驟的版本需要格外的注意,一旦三個自己的版本互相不相容或者和電腦的顯示卡不相容,就用不了。所以安裝前看看要安裝的tensorflow版本。先查好顯示卡的算力,然後適配的相應版本再安裝,否則踩坑要很久。

首先確定電腦所能支援的tensorflow版本,根據tensorflow官網所給配置,我們要去檢查電腦gpu的cuda支援版本,再去對應下載python版本和tensorflow版本。

右鍵 桌面>NVIDIA控制面板>幫助>系統資訊>元件

Window10上Tensorflow的安裝(CPU和GPU版本)

上圖說明我的顯示卡所支援的cuda版本為11.1(向下相容)

Tensorflow配置window官網:https://tensorflow.google.cn/install/source_windows

Window10上Tensorflow的安裝(CPU和GPU版本)

以我電腦為例,之前我用的是3.7.3的版本。這次重灌系統後安裝的是3.8.3。

cmd檢視python版本:Window10上Tensorflow的安裝(CPU和GPU版本)

我之前tensorflow用的是1.13的版本,這次因為已經安裝了python3.8,因此打算安裝2.0版本。如果仍需要低版本需要重新安裝python較低的版本或者搭一個虛擬環境(但不是很建議)。

第一步,安裝anaconda

anaconda會對應安裝python環境,不一定要最新的, 最新的python版本不一定有相容的cuda加速,有cuda加速也不一定支援電腦的顯示卡。

官網安裝:https://www.anaconda.com/products/individual,之前版本如果官網沒有可能需要找資源。

第二步,安裝cuda工具包

(官網:https://developer.nvidia.com/zh-cn/cuda-toolkit)

CUDA是 NVIDIA 專為圖形處理單元 (GPU) 上的通用計算開發的平行計算平臺和程式設計模型。藉助 CUDA,開發者能夠利用 GPU 的強大效能顯著加速計算應用。在經 GPU 加速的應用中,工作負載的序列部分在 CPU 上執行,且 CPU 已針對單執行緒效能進行優化,而應用的計算密集型部分則以並行方式在數千個 GPU 核心上執行。使用 CUDA 時,開發者使用主流語言(如 C、C++、Fortran、Python 和 MATLAB)進行程式設計,並通過擴充套件程式以幾個基本關鍵字的形式來表示並行性。由於tensorflow最高版本對應的是cuda10.1版本,那下載cuda10.1即可。安裝包連結:https://developer.nvidia.com/cuda-toolkit-archive

Window10上Tensorflow的安裝(CPU和GPU版本)

Window10上Tensorflow的安裝(CPU和GPU版本)

Window10上Tensorflow的安裝(CPU和GPU版本)

CUDA的下載需要掛載VPN,否則下下來只有1kb (好像後面又不一定需要VPN,可以直接下載試試)

自定義安裝,安裝的東西全選了,儘量不要改安裝位置

Window10上Tensorflow的安裝(CPU和GPU版本)

安裝完之後安裝cuDNN,cuDNN是用於深度神經網路的GPU加速庫.注意安裝的版本,選擇自身cuda版本對應的cudnn下載。

官網:https://developer.nvidia.com/rdp/cudnn-archive ,cuDNN的下載需要註冊官網的賬號

Window10上Tensorflow的安裝(CPU和GPU版本)

下載好的cudnn檔案解壓後,將資料夾內的檔案放到cuda對應資料夾下,注意:是資料夾內的檔案,而不要直接複製替換資料夾

Window10上Tensorflow的安裝(CPU和GPU版本)

第三步,安裝tensorflow—gpu

在anaconda prompt裡面安裝:pip install tensorflow-gpu , 後面可以指定版本號,下載慢掛載一個映象源 -i https://pypi.tuna.tsinghua.edu.cn/simple

Window10上Tensorflow的安裝(CPU和GPU版本)

安裝完成:

Window10上Tensorflow的安裝(CPU和GPU版本)

測試是否安裝成功:

import tensorflow as tf
tf.test.is_gpu_available()

Window10上Tensorflow的安裝(CPU和GPU版本)

Window10上Tensorflow的安裝(CPU和GPU版本)

True即表示安裝成功了。

檢視版本號:Window10上Tensorflow的安裝(CPU和GPU版本)

至此安裝成功,可以跑一個小程式測試一下。

import tensorflow as tf
import os
os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
 
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
print('GPU:',tf.test.is_gpu_available())

顯示“GPU True”,也即代表GPU版本安裝成功。

經過幾天調了一下程式碼之後發現tensorflow2相對與1還是有挺多改動的,1裡面能夠執行的程式碼可能2裡面需要一定的修改。

關於Tensorflow2和1上面keras的一些區別可以搜一下相關的資料。

到此這篇關於Window10上Tensorflow的安裝(CPU和GPU版本)的文章就介紹到這了,更多相關Window10安裝Tensorflow 內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!