168 字
1 分鐘
Conda 安裝 TensorFlow GPU

介紹#

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 打開./tensorflow-gpu.jpg

Conda 安裝 TensorFlow GPU
https://aishuafei.com/posts/conda-安裝-tensorflow-gpu/
作者
愛耍廢
發佈於
2020-04-10
許可協議
CC BY-NC-SA 4.0