environmental set-up

I am trying to run your code with mnist data. It might be a dumb question, sorry in advance, I just follow the regular caffe2 conda installation and create my own environment! The code tries to use caffe2 within my environment how can I replace it with your version of caffe2?