tensorflow-gpu

win7 cuda-9.0 cudnn7 tensorflow-gpu 1.5配置

這幾天一直在配置tensorflow-gpu版本,之前用的是tensorflow-cpu版本,訓練過比較簡單的mnist手寫數字識別模型,也直接跑過別人訓練好的openpose人體姿態識別模型,效率都比較低,既然想要繼續學習計算機視覺這一塊,最好是能用gpu,可以大大節省模型訓練時間。這幾天遇到不少 […]

win10 安裝cuda9.0 cuDNN7 tensorflow-gpu1.4

在踩過幾乎所有的坑之後,終於把tensorflow 安裝好了,記錄一下我的步驟及注意事項: 首先是檢查GPU的型號是不是cuda支援的型別,點選開啟連結,不過這個列表裡面好像沒有及時更新,我的是GPU型號是GeForce 940MX 裡面沒有列出來,也是支援的。 其次是先安裝VS2015再安裝cud […]

【深度學習】Ubuntu16.04 cuda8.0 cudnn6.0 Python3.6 Anaconda tensorflow-gpu1.3配置(踩坑記)

前言 最近換了帶GPU的新電腦,終於有機會體驗深度學習,然而搭個環境搭得我嘔心瀝血,各種莫名其妙的ERROR,差點就從入門到放棄……好不容易終於成功了,記錄一下QAQ 安裝顯示卡驅動 檢查自己的顯示卡支不支援GPU加速計算,也就是CUDA功能,這個可以在CUDA官網上找到; 在Nvidia官網查詢自 […]