1. 程式人生 > >關於安裝tensorflow-gpu

關於安裝tensorflow-gpu

       安裝tensorflow-gpu重點一定型號匹配。我的顯示卡是NVIDIA Quadro P4000。軟體安裝的是CUAD9.0  CUDNN9.0  python3.6.5

tensorflow-gpu 1.8.0 。

        1、首先安裝python,網上一大堆就不細說了。顯示卡驅動記得安裝!!!!

        2、CUDA下載地址:https://developer.nvidia.com/cuda-toolkit-archive  ,下載的時候選對版本和型號,我下載的cuda9.0,然後安裝到電腦上。在電腦環境path里加入你cuda安裝路徑下的bin路徑 和lin\x64路徑。

將這三個資料夾裡的檔案複製到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0 (如果你安裝cuda9.0的時候路徑沒有改過的話是這個路徑,如果改動了,找到NVIDIA GPU Computing Toolkit資料夾路徑,再往下找cuda v9.0)

將檔案複製到相應的位置。如bin的檔案複製到這個bin下。

       4、複製完成後安裝tensonflow-gpu。 在cmd裡執行 pip install tensorflow-gpu==1.8.0 安裝完後測試一下。

進入到python命令列中, import tensorflow即可。若是不報錯說明沒問題了。若是不放心,網上找個測試例子。

    ps:若是報錯說缺少 某dll檔案,根據提示的錯誤資訊解決。我當時因為缺少的是編譯c++的環境,去微軟官網下載Microsoft Visual C++元件,根據系統版本選擇。