关注很久的一个模型今天开源,欣喜若狂的clone下来测试一下效果,但是它是基于Pytorch的,之前安装过的是TF和caffe,所以刚好搭一下Pytorch的环境。
首先去Pytorch官网,选择配置,这里选择conda安装方式,但是发现它默认给我装的是Python3.7,但是我需要的是Python2.7。所以先用conda创建并激活一个Python2.7的虚拟环境:
conda create -n py27 python=2.7
source activate py27
在python2.7虚拟环境中运行命令:
conda install pytorch torchvision cudatoolkit=8.0 -c pytorch
接下来就是耐心的等待,下载速度蛮慢的。安装成功后简单验证一下:
成功,Pytorch的安装还是相对简单的。