zoukankan      html  css  js  c++  java
  • 解决pytorch报错ImportError: numpy.core.multiarray failed to import

    项目场景

    最近在学pytorch,先用官网提供的conda命令安装了一下:

    conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
    

    然后按照官网的方法测试是否安装成功,能不能正常使用:

    import torch
    x = torch.rand(5, 3)
    print(x)
    

    若能正常打印出x的值,则说明可以正常使用,否则反之。

    问题描述

    代码运行报错:ImportError: numpy.core.multiarray failed to import

    D:miniconda3envspytorchlibsite-packages
    umpy\__init__.py:138: UserWarning: mkl-service package failed to import, therefore Intel(R) MKL 
    initialization ensuring its correct out-of-the box operation under condition when Gnu OpenMP had already been loaded by Python process is not 
      from . import _distributor_init
    Traceback (most recent call last):
      File "c:/Users/ghgxj/Desktop/pytorch/1_get_started.py", line 1, in <module>
        import torch
      File "D:miniconda3envspytorchlibsite-packages	orch\__init__.py", line 189, in <module>
        from torch._C import *
    ImportError: numpy.core.multiarray failed to import
    

    原因分析

    python环境自带一个numpy,会与安装的pytorch带的numpy版本间发生冲突。

    解决方案

    1. 先卸载numpy
    conda uninstall numpy
    
    1. 再重装numpy
    conda install numpy
    

    最终结果

    按照上述方案处理后正常打印出了x的值:

    tensor([[0.8338, 0.1541, 0.0379],
            [0.4348, 0.0145, 0.3586],
            [0.4098, 0.2363, 0.5405],
            [0.7372, 0.7418, 0.3703],
            [0.5668, 0.9512, 0.8041]])
    

    我们再来看看GPU驱动和CUDA是否能用:

    import torch
    print(torch.cuda.is_available())
    

    控制台打印显示True,则说明能正常使用。

    引用参考

    https://pytorch.org/get-started/locally/
    https://blog.csdn.net/wgx571859177/article/details/78273764
    
  • 相关阅读:
    二维码生成:使用 JavaScript 库QRCode.js生成二维码
    VC++6.0远程调试(亲试成功)
    Linux同时安装python2和Python3
    Python打包-py2exe
    camera按键采集图像及waitKey的用法(转)
    ucos实时操作系统学习笔记——任务间通信(信号量)(转)
    STM32硬件IIC驱动设计(转)
    基于STM32F4移植W5500官方驱动库ioLibrary_Driver(转)
    USB基础知识概论(版本:v0.9.2)
    usb帧格式
  • 原文地址:https://www.cnblogs.com/ghgxj/p/14219132.html
Copyright © 2011-2022 走看看