1. 创建虚拟环境pytorch
打开anaconda prompt,输入
conda create --name pytorch python=3.7 anaconda
(最后加anaconda是为了在这个虚拟环境中把numpy, jupyter, pandas等一起安装了,所以耗时较长,也可以不加)
2. 进入虚拟环境
输入
conda activate pytorch
3. 安装pytorch+torchvision
打开torch官网 https://pytorch.org/,根据自己电脑的配置选择相应选项,得到代码
第三行出现了包括conda, pip, source等多种安装方式,根据我多次失败经验,建议不要挣扎,直接选择pip安装。
接下来,大多数教程会告诉你直接把下面的代码复制到prompt就可以等待最后的测试了。
道理是这样没错,但大多数情况下会因为网络不稳定报错,国外的源嘛,这也是为什么大家会选择镜像源的原因。
但我不要(会)!
我就是要用官网下载!
接下来是本教程的核心:
根据官网代码可以得到,官方给我推荐的是1.4.0的torch+0.5.0的torchvision的套餐,连在一起下载耗时长容易报错我分开还不行吗?
进入代码后面的下载网址:https://download.pytorch.org/whl/torch_stable.html
找到对应的安装包,右键复制链接
再在prompt中输入
pip3 install https://download.pytorch.org/whl/cpu/torch-1.4.0%2Bcpu-cp37-cp37m-win_amd64.whl
pip3 install https://download.pytorch.org/whl/cpu/torchvision-0.5.0%2Bcpu-cp37-cp37m-win_amd64.whl
很快安装完成
(如果装到一半失败可以多试几次,会在前面的基础上继续安装)
4. 测试
(pytorch) C:\pythonwork>python
Python 3.7.4 (default, Aug 9 2019, 18:34:13) [MSC v.1915 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
>>> a = torch.ones(5)
>>> a
tensor([1., 1., 1., 1., 1.])