Opencv 3.2.0 with CUDA 8.0 on tensorflow docker
638 ワード
nvidia driver,CUDA 8.0,CUDN 5.1,docker,nvidia-dockerを入れたとします
まず公式のtensorflow dockerバックグラウンド実行を開始し、tagが重要であることを覚えておいてください.その後、GPUを正しく使用するには
containerに進む
GPUをチェック
意外なことにGPUの情報を見ることができます
次にopenCVをインストールし、1行で完了します.
gistここで、scriptの内容についてはOpencv 3.2を参照してください.0 with CUDA 8.0 on Ubuntu 16.04
まず公式のtensorflow dockerバックグラウンド実行を開始し、tagが重要であることを覚えておいてください.その後、GPUを正しく使用するには
:latest-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をインストールし、1行で完了します.
bash
gistここで、scriptの内容についてはOpencv 3.2を参照してください.0 with CUDA 8.0 on Ubuntu 16.04