pytorch的安装

Posted 2020-12-17 juluwangshier

直接使用conda 或者pip 安装都是速度很慢导致安装失败。

参考了下面的博客可以快速安装。

参考：https://www.cnblogs.com/ttzz/p/12347415.html

最后使用先在清华镜像下下载需要的版本，https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

然后将下载的tar包复制到anaconda3/pkgs目录下，

conda install --use-local your-pkg-name

其中 your-pkg-name就是刚才下载的包，比如我下载的是pytorch-1.5.1-py3.6_cuda102_cudnn7_0.tar.bz2

进入到anaconda3/pkgs目录下

安装时就执行：

conda install --use-local pytorch-1.5.1-py3.6_cuda102_cudnn7_0.tar.bz2

 torchvision安装类似：

conda install torchvision-0.6.1-py36_cu102.tar.bz2

 后面就是验证了

import torch

如果不报错说明安装通过。

我遇到的情况是提示numpy错误，

(pytorch) D:Anaconda3pkgs>python
Python 3.6.2 |Continuum Analytics, Inc.| (default, Jul 20 2017, 12:30:02) [MSC v.1900 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "D:Anaconda3envspytorchlibsite-packages	orch\\__init__.py", line 136, in <module>
    from torch._C import *
ImportError: numpy.core.multiarray failed to import

因为我是在新建的虚拟环境安装的，所以其他安装包还没有安装,所以就用命令安装一下就可以：

conda install numpy

然后会提示更新一些其他安装包，安装很快就完成，再次执行

(pytorch) D:Anaconda3pkgs>python
Python 3.6.10 |Anaconda, Inc.| (default, May  7 2020, 19:46:08) [MSC v.1916 64 bit (AMD64)] on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> import torchvision
>>>

这次就不报错了。