因项目需要安装低版本的pytorch=0.4.1
。根据Previous PyTorch Versions进行安装(注意貌似python=2.7
不可行,而且请预先安装好cuda 8.0)
conda create -n pwc python=3.7
conda activate pwc
conda install pytorch=0.4.1 cuda80 -c pytorch
接下来安装torchvision
(注意指定其版本,不然会装成最新版)。
pip install torchvision==0.2