【Bug-驯化】成功解决Some module may need to rebuild instead e.g. with ‘pybind11>=2.12‘.
本次修炼方法请往下查看
🌈 欢迎莅临我的个人主页 👈这里是我工作、学习、实践 IT领域、真诚分享 踩坑集合,智慧小天地!
🎇 免费获取相关内容文档关注:微信公众号,发送 pandas 即可获取
🎇 相关内容视频讲解 B站
🎓 博主简介:AI算法驯化师,混迹多个大厂搜索、推荐、广告、数据分析、数据挖掘岗位 个人申请专利40+,熟练掌握机器、深度学习等各类应用算法原理和项目实战经验。
🔧 技术专长: 在机器学习、搜索、广告、推荐、CV、NLP、多模态、数据分析等算法相关领域有丰富的项目实战经验。已累计为求职、科研、学习等需求提供近千次有偿|无偿定制化服务,助力多位小伙伴在学习、求职、工作上少走弯路、提高效率,近一年好评率100% 。
📝 博客风采: 积极分享关于机器学习、深度学习、数据分析、NLP、PyTorch、Python、Linux、工作、项目总结相关的实用内容。
🎯 1. 问题介绍
我们在安装pytroch版本2.20时,使用的是python3.10版本,这个时候会自动给安装商numpy=2.0.0,这个版本目前来说在运行导入torch时会出现如下问题:
Traceback (most recent call last): File "<stdin>", line 1, in <module>
File "conda/envs/intervl/lib/python3.10/site-packages/torch/__init__.py", line 1471, in <module>
from .functional import * # noqa: F403
File "conda/envs/intervl/lib/python3.10/site-packages/torch/functional.py", line 9, in <module>
import torch.nn.functional as F
File "conda/envs/intervl/lib/python3.10/site-packages/torch/nn/__init__.py", line 1, in <module>
from .modules import * # noqa: F403
File "conda/envs/intervl/lib/python3.10/site-packages/torch/nn/modules/__init__.py", line 35, in <module>
from .transformer import TransformerEncoder, TransformerDecoder, \
File "conda/envs/intervl/lib/python3.10/site-packages/torch/nn/modules/transformer.py", line 20, in <module>
device: torch.device = torch.device(torch._C._get_default_device()), # torch.device('cpu'),
conda/envs/intervl/lib/python3.10/site-packages/torch/nn/modules/transformer.py:20: UserWarning: Failed to initialize NumPy: _ARRAY_API not found (Triggered internally at /opt/conda/conda-bld/pytorch_1704987280714/work/torch/csrc/utils/tensor_numpy.cpp:84.)
device: torch.device = torch.device(torch._C._get_default_device()), # torch.device('cpu'),
💡 2. 解决方法
导致上述的问题的根源就是安装了numpy的高版本,但是目前来说该版本不是支持的特别好,因此,建议使用低版本,这样也不会影响torch的使用,具体的操作如下所示:
pip install numpy==1.26.4
将上述的包安装成功后,导入torch,即可正常,具体的命令如下所示:
>>> import torch
>>> torch.cuda.is_available()
True
>>> quit()
原文地址:https://blog.csdn.net/lov1993/article/details/140460577
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!