Please enable Javascript to view the contents

Conda 安裝 TensorFlow GPU

 ·   ·  ☕ 1 分鐘  ·  ✍️ JK

介紹

GPU計算能力比較快,在Conda上安裝TensorFlow GPU時遇到一些阻礙,一直無法抓到GPU,不要使用pip安裝套件,以下解決方法。

安裝Anaconda

下載網址:https://www.anaconda.com/distribution/

前置作業

需要安裝CUDA+ cuDNN
google關鍵字 CUDA+ cuDNN 可以找到教學資料。

在Conda上安裝套件指令

這個指令會自動下載相關,如果有安裝Keras 會詢問更新適合的版本

conda install tensorflow-gpu

如何降TensorFlow版本

在Winodws 10 作業系統上使用,搜尋 Anaconda Navigator 打開

分享

JK
作者
JK
學習筆記