tensorflow 不調(diào)用顯卡 win10 安裝GPU版本tensorflow

換了一個好顯卡,訓(xùn)練模型的時候比AWS租的慢幾十倍??
看看這一輪訓(xùn)練的時間,長到天荒地老。


image.png

這簡直是一件不能接受的事,難道錢打水漂了?
No,我們 too young too simple!

先使用 nvidia-smi 命令檢查一下。
'nvidia-smi' 命令可以在linux下直接使用,在windows下直接執(zhí)行會報錯:

image.png

直接解決辦法是到 'NVSMI' 目錄下去執(zhí)行,通常目錄位置為:
C:\Program Files\NVIDIA Corporation\NVSMI
執(zhí)行結(jié)果如下:

image.png

看吧, Process name里面沒有一個python, 說明根本沒有調(diào)用顯卡來進(jìn)行計算。

  1. 先卸載目前的tensorlfow :
    pip uninstall tensorflow -y
  2. 然后安裝GPU版本的tensorflow:
    pip install tensorflow-gpu -UI
    如果pip下載實在是太慢,或者容易掉線的話,建議直接復(fù)制地址用工具下載,速度杠杠
    下載whl文件,安裝只需要到文件所在位置,執(zhí)行
    pip install tensorflow_gpu-1.10.0-cp36-cp36m-win_amd64.whl

安裝結(jié)束! 運行試一下,提示錯誤就來了-_-!!~:


image.png

去下載了windows版本

image.png

下載bese installer 以及 4個 patches
https://developer.nvidia.com/cuda-90-download-archive

在這之前要記得安裝NVIDIA驅(qū)動,地址為:
https://www.nvidia.cn/Download/index.aspx?lang=cn

以上成功安裝完成后 , 不出意外的還有問題(喝口水平靜一下)

image.png

下載地址:
https://developer.nvidia.com/cudnn
這個下載需要注冊,驗證郵件以及填一下表格。
下載后得到三個文件夾 ,需要復(fù)制到路徑中。
image.png

查找路徑可以輸入命令:
echo %path%

image.png

找到本機(jī)的CUDA地址,我的是
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
三個文件夾復(fù)制過去。再調(diào)用一下

image.png

表示------‘呵呵‘
升級 numpy / 卸載 unmpy 再重裝:
pip uninstall numpy
pip install numpy --upgrade
numpy 的版本需要注意一下

image.png

需要 numpy<=1.14.5,>=1.13.3,

再運行 nvidia-smi 看看


image.png

運行速度改變了


image.png

感謝各位的閱讀,如對你有幫助,請點喜歡。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容