win7 64位系統tensorflow gpu 1.4安裝指南

之前一直用的tensorflow cpu版的,後來數據量大一些之後,覺得有點算不動了,所以打算改用gpu版。

傳聞gpu不太容易安裝,誰知道這麼坑!

光系統就重裝了3遍,當然這裏面是我自己有點強迫症,最後感覺能否安裝成功與系統關係不大。下面是正經內容。

系統:cn_windows_7_ultimate_with_sp1_x64_dvd_u_677408(這個不重要)

筆記本配置:處理器Intel(R) Core(TM) i7-4710MQ CPU @2.50GHz,內存16G,顯卡NVIDIA GeForce GTX 860M

python版本:3.5.2

tensorflow版本:tensorflow_gpu-1.4.0-cp35-cp35m-win_amd64.whl

安裝過程:

1. 先安裝Python3.5.2,記得添加路徑,這個大家應該很熟了,不做過多介紹。

2. 安裝Visual C++ Redistributable for Visual Studio 2015(https://www.microsoft.com/en-us/download/details.aspx?id=48145),或者直接安裝Visual Studio 2015。

3.安裝CUDA和cudnn

CUDA版本:cuda v8.0.60(cuda_8.0.61_windows.exe和cuda_8.0.61.2_windows.exe,注意有兩個文件,一個是主文件,一個是補丁,都要裝上)

cudnn版本:cudnn v6.0(cudnn-8.0-windows7-x64-v6.0)

先安裝CUDA,先裝cuda_8.0.61_windows.exe,再裝cuda_8.0.61.2_windows.exe(補丁),然後解壓cudnn-8.0-windows7-x64-v6.0,將解壓得到的3個文件夾(bin, include, lib)複製到CUDA的安裝路徑(以我爲例,複製到:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0)

這樣CUDA就安裝好了。想要查看一下的話,打開cmd,輸入nvcc -V,會得到如下信息,

4. 安裝tensorflow gpu 1.4

這裏提示最好聯網安裝,我是將tensorflow_gpu-1.4.0-cp35-cp35m-win_amd64.whl下載到本地,採用pip安裝的,在這過程中,需要什麼別的包,本地沒有,就自動從網上下載了。

我還沒有試過離線安裝,但是cpu版的離線安裝我做過。離線安裝的話,安裝過程中,缺少什麼包會有提示,根據提示先裝上這些擴展包之後再重新安裝tensorflow即可。

5. 安裝好之後,在python中import tensorflow不報錯,但是執行session的時候會報錯,類似於“failed call to cuInit: CUDA_ERROR_NO_DEVICE”或者“設備是不可移動的,無法彈出或拔出問題”之類的錯誤,這個原因是顯卡驅動版本過老引起的。

CUDA安裝過程中自帶一個版本的顯卡驅動376.51,根據下表是符合要求的,但是實際上在我這裏報錯了。最後單獨安裝了一次顯卡驅動,程序運行成功。

來源:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

我的顯卡驅動信息如下:

示例程序:

import tensorflow as tf
import os
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print (sess.run(hello))
a = tf.constant(10)
b = tf.constant(32)
print(sess.run(a+b))

最後程序運行結果如下:

好像看不太清……

複製一下是這樣的:

"C:\Users\ZhuXiaobin\venv\LRE FDD\Scripts\python.exe" "F:/about phd/LRE FDD/測試文件.py"
2019-04-02 17:04:13.172967: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\platform\cpu_feature_guard.cc:137] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX AVX2
2019-04-02 17:04:13.878973: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\common_runtime\gpu\gpu_device.cc:1030] Found device 0 with properties: 
name: GeForce GTX 860M major: 5 minor: 0 memoryClockRate(GHz): 0.54
pciBusID: 0000:01:00.0
totalMemory: 2.00GiB freeMemory: 1.91GiB
2019-04-02 17:04:13.878973: I C:\tf_jenkins\home\workspace\rel-win\M\windows-gpu\PY\35\tensorflow\core\common_runtime\gpu\gpu_device.cc:1120] Creating TensorFlow device (/device:GPU:0) -> (device: 0, name: GeForce GTX 860M, pci bus id: 0000:01:00.0, compute capability: 5.0)
b'Hello, TensorFlow!'
42

Process finished with exit code 0
 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章