Windows下安裝TensorFlow-gpu

在以前須要安裝CUDA Toolkit 8.0.
https://pan.baidu.com/s/1c2y8xIK和cuDNN v6 or v6.1。
必定要版本正確,不然會出現問題,在官網上有說明。python

直接一直點擊確認便可,安裝後能夠看到系統中多了CUDA_PATH和CUDA_PATH_V8_0兩個環境變量,接下來要在系統中添加如下幾個環境變量:   CUDA_SDK_PATH=C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0   CUDA_LIB_PATH=%CUDA_PATH%\lib\x64

   CUDA_BIN_PATH=%CUDA_PATH%\bin
   CUDA_SDK_BIN_PATH=%CUDA_SDK_PATH%\bin\win64

   CUDA_SDK_LIB_PATH=%CUDA_SDK_PATH%\common\lib\x64
   而後,在系統變量PATH的末尾依次添加:
   %CUDA_LIB_PATH%  %CUDA_BIN_PATH%  %CUDA_SDK_LIB_PATH%  %CUDA_SDK_BIN_PATH%  說明:這裏注意的是CUDA_SDK_PATH路徑的位置,不少人去這個錯誤路徑「C:\Program Files\NVIDIA Corporation」下,發現找不到CUDA Samples文件夾。其實是由於ProgramData是一個隱藏文件夾,須要設置【個人電腦】-【查看】-【顯示隱藏文件】才能看到,所以這個纔是正確路徑「C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0」。
  配置成功後,重啓計算機使環境變量生效。

conda create -n tensorflow python=3.5
activate tensorflow
pip install –ignore-installed –upgrade tensorflow-gpuide

相關文章
相關標籤/搜索