配置pytorch环境

安装anaconda

wget https://repo.anaconda.com/archive/Anaconda3-5.1.0-Linux-x86_64.sh

bash Anaconda3-5.1.0-Linux-x86_64.sh

安装pytorch

python3 -m conda install pytorch=0.3.0 torchvision cuda90 -c pytorch

安装tensorboardx

python3 -m conda install -c conda-forge tensorboardx 

安装opencv

python3 -m pip install opencv-python

或

pip install opencv-contrib-python

如果要使用带孔的densenet，安装这个

http://hangzh.com/PyTorch-Encoding/notes/compile.html

git clone https://github.com/zhanghang1989/PyTorch-Encoding && cd PyTorch-Encoding

pip install -r requirements.txt
python setup.py install