2018年1月23日 星期二

Tensorflow 1.4 gpu版 安裝步驟

0.安裝Pycharm 2017.1.5

1.安裝cuda_8.0.61_win10.exe

2.下載cudnn-8.0-windows10-x64-v6.0,
將bin, lib, include裡面檔案放到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\xxx

3.安裝Anaconda3-4.1.1-Windows-x86_64.exe
安裝到C:\Anaconda3

4.Anaconda Prompt (or CMD)
conda create -n tensorflow-gpu python=3.5.2
activate tensorflow-gpu (activate後的動作都會安裝到C:\Anaconda3\envs\tensorflow-gpu裡面的python套件管理)
pip install tensorflow-gpu keras pillow h5py (如果有缺套件一樣從這邊新增, 可用pip list檢查)

5.pycharm new project
Interpreter : C:\Anaconda3\envs\tensorflow-gpu\python.exe (應該會需要一點時間,pycharm右下角 Background Task)





# 參考網址 http://blog.nitishmutha.com/tensorflow/2017/01/22/TensorFlow-with-gpu-for-windows.html

# 版本需對應 cuda8 + cuDNN v6

沒有留言:

張貼留言