Build the Dockerfile using docker build -t venkai/nvcaffe . If you have an older NVIDIA driver that you don't want to upgrade, you can modify the first line of the Dockerfile with the appropriate nvidia/cuda image using the compatibility table provided here.