Windows10 Tensorflow安裝、測試爬坑指南

最近在看莫煩的Tensorflow(乾貨必須安利一波,https://www.bilibili.com/video/av16001891/?p=9)

想自己通過Tensorflow搭建一個AutoEncoder,於是開始安裝Tensorflow-Gpu版,遇到各種坑,終於在不斷搜索各種問題的解決辦法終於爬出了坑,現整理、總結爬坑指南如下:

首先說明我是配置Tensorflow-gpu + CUDA9.1 + cudaNN for CUDA9.1.

1.安裝CUDAv.9.1

進入NVIDIA下載CUDA頁面(https://developer.nvidia.com/cuda-downloads),下載。根據自己PC/Server配置下載對應版本。建議下載Local版本地安裝。

按照引導不斷Next,安裝成功後你會發現生成的文件夾:

2.安裝cuDNN

先要在NVIDIA官網註冊個賬號,然後填信息,I agree ...,這個搞定後 就可以開始下載cuDNN了。注意版本!!!


下載完成解壓將三個文件夾複製並粘貼到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1目錄下。

PS:最好將bin和lib\x64加到環境變量path中,不加可能後邊操作會報錯。

3.安裝Tensorflow

之前看了一些帖子說CUDAv9.1 太新用不了tnsorflow,苦苦尋找終於找到一個大神帖子(https://blog.csdn.net/qq_36556893/article/details/79433298)抱着試一試的態度,成功!!!感謝大神。

Anaconda中創建環境和激活環境就不說了。。。,我主要說下在環境中安裝tensorflow。

如果環境中選擇直接輸入指令:pip install --ignore-installed --upgrade tensorflow,安裝不會報錯。

But在import tensorflow as tf會報錯,顯示OSError: [WinError XXX] 找不到指定的模塊/Could not find 'cudart64_90.dll'.這是因爲目前的Tensorflow最多支持到CUDA9.0,絕望!然而,大神告訴你   --->

一個大神編譯的.whl文件,鏈接:https://pan.baidu.com/s/1dGSmo29 密碼:fwet

下載完成後請複製到本地用戶目錄文件夾底下,例如:

繼續輸入指令:pip install tensorflow_gpu-1.6.0-cp36-cp36m-win_amd64.wh,安裝成功!!!

4.測試Tensorflow

你可以在環境下安裝jupyter或者直接python進入編輯環境測試。

輸入以下指令:

import tensorflow as tf

hello = tf.constant('My,TensorFlow!')

sess = tf.Session()

print(sess.run(hello))

最後一行顯示 b'My,TensorFlow!'說明Tensorflow配置成功!!

PS:用jupyter測試TensorFlow步驟,附上峯大神的Anacnda管理乾貨(https://blog.csdn.net/qq_24375467/article/details/72486476

     

 


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