Tensorflow-gpu 安裝教學

陳怡中
2 min readAug 1, 2018

--

由於近年來GPU的進步以及軟體的跟上,於是深度學習運算效率大量提升,若要進行深度學習的運算建議買一張Nvidia顯卡

1. 購買顯卡

在買顯卡前建議先參考此網站https://itw01.com/VAUKEOI.html,除此之外還要再去Nvidia官方網站確認該顯卡是否有支援深度學習,根據自己的需求去決定顯卡,接著再去原價屋購買顯卡。接著等二~三天等電腦送來即可。

2. 下載驅動程式

(1)在Nvidia的網站下載驅動程式CUDA
https://developer.nvidia.com/cuda-downloads

(2)在Nvidia網站下載另外一個驅動程式cudnn64_5.dll(免費下載,不過要登入)
https://developer.nvidia.com/cudnn

(3)CUDA安裝

下一步到底,BJ4

(4) 設定環境變數
Windows 介面:
編輯系統環境變數 > 進階 > 環境變數 > 新增 > 增加變數名稱(Path) >
輸入路徑 > 將cudnn放到此路徑

(5) 安裝tensorflow
在cmd介面中打上 pip install tensorflow-gpu

(6) 測試安裝是否成功
打開cmd 輸入ipython, 輸入 import tensorflow as tf ,若沒出錯及安裝成功。

Sign up to discover human stories that deepen your understanding of the world.

Free

Distraction-free reading. No ads.

Organize your knowledge with lists and highlights.

Tell your story. Find your audience.

Membership

Read member-only stories

Support writers you read most

Earn money for your writing

Listen to audio narrations

Read offline with the Medium app

--

--

陳怡中
陳怡中

Written by 陳怡中

正在AI方向努力的小小工程師

No responses yet

Write a response