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