1. 程式人生 > >[Tensorflow]環境搭建vs2017+win10+py3.6+cuda9.1+cudnn7+tf1.5

[Tensorflow]環境搭建vs2017+win10+py3.6+cuda9.1+cudnn7+tf1.5

一、安裝cuda 9.1+VS2017

一路下一步即可,環境變數cuda會自動配好。

安裝完後,在主目錄搜尋deviceQuery.exe,跑一跑測試下就好了(成功最下一句是:RESULT PASS)。

vs2017和cuda 9.1同時安裝的,雖然cuda暫時提示找不到vs,但是不影響。

二、安裝cudnn7.0

下載後解壓,然後複製檔案:

 3. Copy the following files into the CUDA Toolkit directory.

   a) Copy \cuda\bin\cudnn64_7.dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin.

   b) Copy \cuda\ include\cudnn.h to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include.

   c) Copy \cuda\lib\x64\cudnn.lib to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64.

三、tensorflow安裝

據說tf不支援cuda 9.1(因為官方的wheel是在cuda 9.0上編譯的~)

當然有大神自己編譯了一堆whl,方便不同版本使用。

https://github.com/fo40225/tensorflow-windows-wheel

找到對應cuda9.1版本,tf1.5的whl檔案。

安裝成功。