tensorflow 1.12.0 gpu + python3.6.8 + win10 + GTX1060 + cuda9.0 + cudnn7.4 + vs2017)


 在安裝tensorflow-gpu時,也看過不少的博客,講得亂七八糟的,也不能這樣說,只是每個人安裝的環境或需求不一樣,因此沒有找到一個適合自己的教程去安裝tensorflow-gpu版本。當然,入手一台新電腦立馬是安裝配置這些環境,在次期間也遇到過不少的坑。話不多說,對此總結了以下幾個步驟,直到成功!

1、首先查看tensorflow-gpu所對應的python版本、CUDA、cuDNN版本,這個非常重要,直接決定,最后是否安裝成功。

注:鏈接查看地址https://tensorflow.google.cn/install/source_windows

2、下載CUDA,並且安裝

CUDA(Compute Unified Device Architecture),是顯卡廠商NVIDIA推出的運算平台。

當然在下載時,也遇到過不少尷尬的情景,比如說:下載超時,遇到下載超時的情況,解決辦法:通過網絡版下載並進行安裝,效果是一樣的!本次筆者下載的是CUDA9.0,其下載地址:https://developer.nvidia.com/cuda-toolkit-archive

3、下載cudnn

下載地址:https://developer.nvidia.com/rdp/cudnn-download

注意:下載時需要注冊會員,只需要按照要求,簡單注冊即可。本次筆者,下載的是cuDNN7.4.2版本!

 

下載后,解壓cudnn-9.0-windows10-x64-v7.4.2.24

打開cudnn-9.0-windows10-x64-v7.4.2.24文件夾,得到bin、include、lib三個文件夾

4、將CUDNN中的bin、include、lib文件黏貼到CUDA9.0中

安裝完成cuda9.0過后,將cudnn中的三個文件(bin、include、lib)黏貼到對應的cuda9.0文件中,當然需要准確到找到cuda9.0的安裝位置及其對應的文件。

5、檢測是否安裝好cuda

cmd中輸入nvcc -V命令,如果出現了版本信息,則說明已經安裝好cuda

6、Pycharm中安裝tensorflow-gpu版本

直接在pycharm的terminal中數據:pip install tensorflow-gpu命令即可,筆者安裝的是tensorflow-gpu1.12.0版本

7、測試安裝tensorflow-gpu版本是否成功

新建一個python腳本,進行測試

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

得到輸出為:

恭喜你,已經成功安裝TensorFlow-GPU版本!

注:試錯的過程是漫長的,但是成功那一下是非常有成就的,安裝成功的關鍵,一句總結:多看官網的說明!


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM