Docker build for the deep learning framework