1. 程式人生 > >Win10+GTX 1080Ti+Anaconda TensorFlow安裝

Win10+GTX 1080Ti+Anaconda TensorFlow安裝

實驗室新配了電腦,折騰了一整天,總算是把環境配好了

查看了很多blog,都是說GTX1080Ti目前只能使用cuda v8.0(可以參考Keras的中文文件,寫的非常詳細https://keras-cn.readthedocs.io/en/latest/for_beginners/keras_windows/),當然也有大神嘗試過其他版本(win10+1080Ti+cuda9.1+cudnn7+python3.6),但由於本人不敢再踩坑了,所以有興趣的朋友可以自己試試看啦:)https://blog.csdn.net/u012746060/article/details/81182727

步驟

1、下載cuda v8.0安裝,選擇自定義,只選第一個就行

2、下載cudnn v6.0並解壓

注意:上兩步一定要注意版本,不然安了後面還是要解除安裝重灌的

3、將cudnn下面的三個檔案複製到cuda安裝路徑下對應的資料夾中,這裡有一個可以測試cuda是否安裝成功的方法,參見https://blog.csdn.net/wanglei022/article/details/78659955 使用cmd命令列格式,CD到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0目錄下 找一下 extras\demo_suite 資料夾。裡面提供了NVIDIA自帶的兩個測試.exe檔案:deviceQuery.exe 和 bandwidthTest.exe 。 在命令列輸入檔名(如bandwidthTest.exe),如果看到輸出的最後result = pass 就說明安裝成功! 恭喜你:)。

4、建立TensorFlow環境

  conda create -n tensorflow_gpu python=3.5

  activate tesorflow_gpu

  pip install tensorflow-gpu==1.3

 注意,這裡tensorflow-gpu要選定1.3版本的

5、測試

import tensorflow,如果沒有報錯的話,安裝成功了

安裝期間出現的問題:No module named '_pywrap_tensorflow_internal'

解決方案,注意cudnn和cuda的版本問題,不知道是1080Ti還是win系統的問題,對cudnn和cuda的版本要求很高,真的是差一點都不行。之前嘗試過的版本1)cuda v10.0結果失敗了2)cuda v8.0+cudnn v7.0失敗3)cuda v8.0+cudnn v6.0終於成功了

參考blog

https://blog.csdn.net/he5688/article/details/77542011

https://blog.csdn.net/wanglei022/article/details/78659955

https://keras-cn.readthedocs.io/en/latest/for_beginners/keras_windows/