[Tensorflow]環境搭建vs2017+win10+py3.6+cuda9.1+cudnn7+tf1.5
阿新 • • 發佈:2019-02-12
一、安裝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檔案。
安裝成功。