Opencv 3.2.0 with CUDA 8.0 on tensorflow docker

假設你灌好了nvidia driver, CUDA 8.0, CUDNN 5.1, docker, nvidia-docker

先啟動官方的tensorflow docker背景執行,切記tag很重要,後面加上:latest-gpu才能正確使用GPU,否則預設會用:latest,也就是CPU版本。

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

進到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在此,關於script內容,請參考Opencv 3.2.0 with CUDA 8.0 on Ubuntu 16.04

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容