-
Update NVIDIA driver
-
Installing Dependencies
-
Install CUDA
-
Install cuDNN
-
Install Caffe2
-
Set environment variables
-
Test Caffe2 & GPU
caffe2/caffe2:snapshot-py2-cuda9.0-cudnn7-ubuntu16.04
在caffe2基础上安装facebook的Detectron模块。
-
Install Detectron dependencies
-
Install Detectron
参考文献:
- https://github.com/facebookresearch/Detectron/issues/122
- https://hub.docker.com/r/caffe2/caffe2/
- https://github.com/facebookresearch/Detectron