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