TensorFlow-Gpu 1.8安裝

原創 lightcity 光城 2018-10-21

TensorFlow-Gpu 1.8安裝

0.說在前面

1.Cuda與cuDNN

2.配置及測試

3.tensorflow-gpu 1.8

4.問題

5.做者的話

0.說在前面

終於又到週末了，你們週末快樂，咱們都知道在數據挖掘裏面有個比賽：Kaggle，這兩天在玩Kaggle比賽之泰坦尼克號問題，在下面幾節將會詳細介紹，泰坦尼克號問題思路及Kaggle比賽玩法。咱們一塊兒來期待吧！
這個是當前的成績：

總共有9000多人，排名算了一下在6%到10%之間，線上成績0.8+。

今天主要分享一下TensorFlow新版gpu版本在win10上安裝的問題

版本

• Win10+PyCharm

• Cuda9.2

• cuDNN7.1

• tensorflow-gpu1.8

點擊公衆號右下角合做轉載->聯繫我，便可加入個人我的微信，共同探討交流，以及入交流羣(記得備註入羣)！

1.Cuda與cuDNN

關於Cuda 9.2與cuDNN 7.1的安裝，下載後直接安裝便可！
下面是官網連接！

Cuda官網下載：

https://developer.nvidia.com/cuda-toolkit-archive

cuDNN官網下載(需註冊)：

https://developer.nvidia.com/rdp/cudnn-download

2.配置及測試

關於Cuda 9.2與cuDNN 7.1安裝環境配置及測試
【Cuda 9.2配置】

一路安裝成功後，dos輸入nvcc -V，顯示以下信息表示安裝成功

默認狀況

安裝好後，默認狀況下，系統變量會多出CUDA_PATH和CUDA_PATH_V9_2兩個環境變量。

添加系統變量

CUDA_SDK_PATH = C:\ProgramData\NVIDIACorporation\CUDA Samples\v9.2
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

【cuDNN 7.1配置】

解壓cudnn-9.2-windows10-x64-v7.1.zip，將文件夾裏的內容拷貝到CUDA的安裝目錄並覆蓋相應的文件夾，CUDA默認安裝目錄：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2

3.tensorflow-gpu 1.8

【安裝】

注：目前tensorflow還不能支持cuda9.2，因此只能經過源碼編譯或者別人編譯好的安裝包安裝。

下載地址：

https://github.com/fo40225/tensorflow-windows-wheel/tree/master/1.8.0/py36/GPU/cuda92cudnn71sse2

下載這個wheel包後，使用pip install …..whl進行安裝

【測試】

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
Hello, TensorFlow!

4.問題

警告：Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA

#解決
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

FutureWarning: Conversion of the second argument of issubdtype from float to np.floating is deprecated

h5py新版本對numpy1.4版本的兼容錯誤
更新numpy與h5py問題解決

5.做者的話

最後，您若是以爲本公衆號對您有幫助，歡迎您多多支持，轉發，謝謝！

更多內容，請關注本公衆號深度學習系列！