Caffe2 is a lightweight, modular, and scalable deep learning framework.
conda install caffe2::caffe2-cuda-cudnn-full