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
沒有留言:
張貼留言