【那些年踩过的坑】win10 + python 3.6 环境下安装 pytorch/torchvision

我的安装环境和版本

  • Windows 10
  • Python 3.6
  • Anaconda 3 64-bit
  • Pytorch CPU version
  • torchvision

安装步骤

  • 安装python3.6/3.5(个人用的是3.6。3.5也可以,但我没试过)
  • 安装anaconda3 64-bit
    注意:

    1. 一定要看好型号哇,windows下pytorch仅支持3.5和3.6,所以只能安装anaconda3。并且最好安装64位的。我一开始安装32位的版本,结果安装pytorch包时,报错package not found,可能是没有32位的包
    2. 安装anaconda时推荐选“only install for one user”,最好别选“install for all users”。很多选择了multi-user的用户都出现了安装/更新包时报错permission denied.
  • 在官网上选择合适的安装方式
    https://pytorch.org/
    【那些年踩过的坑】win10 + python 3.6 环境下安装 pytorch/torchvision_第1张图片
    我的选择如上图所示。

  • 成功运行完第一行命令后,痛苦又开始了!
    我用pip3安装torchvision竟然报错!
    【那些年踩过的坑】win10 + python 3.6 环境下安装 pytorch/torchvision_第2张图片
    报错里建议我用conda安装,然后尝试了各种“conda install xxx torchvision xxx”,均以失败告终。我甚至尝试了下载whl文件直接安装,都没成功。
    最后,我竟然鬼使神差地用了pip而不是pip3,安装成功了!如果你同时安装了python2.x,不建议尝试pip安装,容易搞混。

pip install torchvision

【那些年踩过的坑】win10 + python 3.6 环境下安装 pytorch/torchvision_第3张图片

  • 最后测试一下是否安装成功
    天真如我竟然输入了 “import pytorch”
    下面才是正确的代码
import torch
import torchvision

你可能感兴趣的:(那些年踩过的坑)