Tensorflow-gpu環境搭建


顯卡設備:英偉達1060

系統:Windows10

python版本:3.7.4

CUDA:cuda_10.0.130_411.31_win10

cudnn:cudnn-10.0-windows10-x64-v7.6.5.32

tensorflow:gpu1.14

一 安裝python3.7.4

  https://www.python.org/downloads/windows/ 下載安裝

二 安裝cuda:

  1.首先更新顯卡驅動;

  2.桌面右鍵-NVIDIA控制面板-幫助-系統信息-組件,可以查看當前設備支持的最高cuda版本。

  我更新驅動之前支持最高到8,更新后就支持10版本了。

  

 

  3.下載cuda

  https://developer.nvidia.com/cuda-downloads 

  安裝。一直下一步就行。選擇組件的時候可以不選擇顯卡驅動,如果你的驅動比較新。

  4.環境變量

  

 

 三 cudnn安裝

  1 下載https://developer.nvidia.com/rdp/cudnn-archive cuda和cudnn有對應版本關系。

  2.解壓文件夾,將解壓后的文件夾下的文件拷貝到cuda安裝目錄下,與之相對應的文件夾下。

    解壓后其實是三個文件夾,合並到cuda的安裝目錄中就可以了。

    默認路徑是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0 找不到了可以類比找一下

四 安裝tensorflow-gpu

  cmd中直接運行

  pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.14

    

 

 


免責聲明!

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



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