码迷,mamicode.com
首页 > 其他好文 > 详细

pytorch的安装

时间:2020-07-11 12:40:57      阅读:217      评论:0      收藏:0      [点我收藏+]

标签:版本   tps   array   报错   back   edit   python   tun   ORC   

直接使用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:\Anaconda3\pkgs>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:\Anaconda3\envs\pytorch\lib\site-packages\torch\__init__.py", line 136, in <module>
    from torch._C import *
ImportError: numpy.core.multiarray failed to import

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

conda install numpy

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

(pytorch) D:\Anaconda3\pkgs>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
>>>

这次就不报错了。

 

pytorch的安装

标签:版本   tps   array   报错   back   edit   python   tun   ORC   

原文地址:https://www.cnblogs.com/juluwangshier/p/13283089.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!