win10_64位下搭建TensorFlow-gpu版本


安裝gpu版本首先是英偉達的顯卡。

需要用到的軟件有:

  1.python 3.6

  2.Tensorflow 1.11

  3.CUDA:cuda_8.0_win10.exe

  4.cuDNN: cudnn-8.0-windows10-x64-v6.0.zip

讓我們開始吧~

 1.首先,下載cuda8.0版本以及cudnn6.0版本(cuda和cudnn的版本一定要對應)。

    cuda8.0下載鏈接:https://developer.nvidia.com/cuda-80-download-archive

      cudnn6.0下載鏈接:https://developer.nvidia.com/rdp/cudnn-archive

 

 

2.讓我們安裝cuda8.0

  點擊運行下載好的cuda8.0.exe。會出現下圖,注意!!!!這個文件夾是安裝過程中的臨時文件存儲的目錄,並不是最終的安裝目錄(默認安裝的主目錄是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0)。后邊就很簡單了,一步步安裝好。

讓我們驗證以下是不是安裝成功了~

  打開cmd,輸入nvcc -V

  如圖所示,我們的cuda已經安裝好了,並且顯示了版本號8.0.6

3.安裝cuDNN6.0

  把cudnn6.0.zip解壓如下圖

  然后找到cuda8.0的主目錄(也就是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0),將解壓出來的三個子文件copy到該目錄下,成功的話能看到cudnn64_6.dll

4.安裝tensorflow(以anaconda為例)

  首先創建tensorflow gpu版本的環境。在anaconda prompt中輸入:

    conda create -n tensorflow-gpu python=3.6

  然后激活:

    activate tensorflow-gpu

  最后就是安裝:

    conda install tensorflow-gpu

等待結束就可以啦。

5.在pycharm中使用

  file->setting 解釋器一定要選anaconda\envs\tensorflow-gpu\python.exe要不在pycharm中會報錯

6.測試一下吧

           import tensorflow as tf

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

 如圖所示,測試程序成功!!!!

tensorflow你就可以用啦。

 

 

 

 

 

 

  

 


免責聲明!

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



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