Note this implementation works with pytorch=0.4.0 and cuda8.0.
conda env create -f csm.yml
source activate csm
mkdir sources
pip install -U pip
pip download cupy==2.3.0
tar -xf cupy-2.3.0.tar.gz
cd cupy-2.3.0
python setup.py install
git clone [email protected]:nileshkulkarni/csm.git csm_root
cd csm_root/csm/external/
sh install.sh
pip install -r docs/requirements.txt