關於安裝tensorflow-gpu
阿新 • • 發佈:2018-12-18
安裝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++元件,根據系統版本選擇。