1.首先创建一个pytorch的环境,
先打开Anaconda Prompt对话框,然后直接输入
conda create -n pytorch python==3.6#这里指定了python的版本,如果不指定则默认创建的最新版本的python
然后激活创建的pytorch环境,在pytorch环境中安装torch和torchvision
conda activate pytorch#之后进入到pytorch环境
在安装torch的时候,直接pip install安装会很慢很慢,这里建议先下载要安装的torch版本,然后再用pip install (下载的torch文件位置)
pip install D:/ProgramFiles/Anaconda/envs/torch-1.1.0-cp36-cp36m-win_amd64.whl#安装torch
此时注意在安装torch的同时会安装numpy,如果numpy安装出现错误则会安装不成功。
检查torch安装是否成功,则在pytorch环境下输入python进入命令行模式,然后输入import torch如果没有错误则证明安装成功。
接下来退出命(exit())令行的模式回到pytorch的环境中安装torchvision,torchvision的安装和torch类似,先下载自己需要安装的版本,然后再在对应的目录进行安装。
torchvision的下载:
https://pypi.org/project/torchvision/0.1.8/
点击网址中左侧的"Release history",可以下载自己想要的版本,注意torchvision的版本对torch的版本又要求。
在安装torchvision的时候会要求安装pillow的版本,在这里我安装的torchvision是0.3.0版本的要求pillow版本大于等于4.1.1,当时我直接pip install pillow的时候默认安装的是pillow的最新版本。如果安装最新版本则在torchvision安装成功后,在测试环境中输入import torchvision会出错,显示找不到此模块。这是因为pillow的高版本和torchvision要求的4.1.1不匹配,因此需要用pillow4.1.1版本,若此时直接再安装pillow4.1.1版本还是不行。因此需要将torchvison卸载后先安装pillow4.1.1才可行。所以最好的方式是再安装torchvision0.3.0之前先安装pillow4.1.1。
pip install pillow==4.1.1#安装pillow
pip install D:/ProgramFiles/Anaconda/envs/torchvision-0.3.0-cp36-cp36m-win_amd64.whl
#安装torchvision0.3.0版本
如下图所示即为安装成功。