Opencv 3.2.0 with CUDA 8.0 on tensorflow docker

假設(shè)你灌好了nvidia driver, CUDA 8.0, CUDNN 5.1, docker, nvidia-docker

先啟動(dòng)官方的tensorflow docker背景執(zhí)行,切記tag很重要,後面加上:latest-gpu才能正確使用GPU,否則預(yù)設(shè)會(huì)用:latest,也就是CPU版本。

sudo nvidia-docker run -d -p 8888:8888 gcr.io/tensorflow/tensorflow:latest-gpu

進(jìn)到container

sudo nvidia-docker exec -it {container ID} bash

檢查一下GPU

nvidia-smi

沒意外的話可以看到GPU的資訊

接著安裝openCV,一行搞定

bash <(curl -s https://gist.githubusercontent.com/red010182/cb8133c1bd4daf41c82803d819ce1784/raw/9dd46c6b9f248ea6a311c6f1c42ef703eac63c04/install_opencv.sh)

gist在此,關(guān)於script內(nèi)容,請(qǐng)參考Opencv 3.2.0 with CUDA 8.0 on Ubuntu 16.04

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

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

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