AttributeError: module 'torch' has no attribute 'asarray' - Jetson TX2

link管理

链接快照平台

输入网页链接，自动生成快照
标签化管理网页链接

相关文章推荐

活泼的面包 · python实现自动化测试报告邮件实时发送 ...· 2 周前 ·

逼格高的作业本 · AIX Toolbox for Open ...· 2 周前 ·

慷慨的小刀 · 重要新闻_中华人民共和国外交部· 2 周前 ·

爱热闹的竹笋 · Failed to apply spec ...· 5 月前 ·

大鼻子的伤疤 · 广州电车_百度百科· 5 月前 ·

腹黑的核桃 · 滴水穿石二十载 ...· 6 月前 ·

越狱的蚂蚁 · Steam 上的 The Balls· 7 月前 ·

Hello,

I’m trying to use docker environment with TX2.

I builded PyTorch and torchvision as below guide:
(Wheel file is builded on JetPack 4.6 / CUDA 10.2 / Python3.8)

PyTorch for Jetson Announcements Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer. Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson. These pip wheels are built for ARM aarch64 architecture, so run these commands on your Jetson (not on a host PC). You can also use the containers from jetson-containers . PyTorch pip wheels JetPack 6

PyTorch v2.2.0

JetPack 6.0 DP (L4T R3…

The verification is OK with our environment.
But some function has a problem as below :

root@tx2:/home/test/YOLOv8-TensorRT# python3 infer-det.py --engine yolov8s.engine --imgs data --show --out-dir outputs --device cuda:0
/usr/local/lib/python3.8/dist-packages/torch/package/_directory_reader.py:17: UserWarning: Failed to initialize NumPy: module compiled against API version 0x10 but this version of numpy is 0xf . Check the section C-API incompatibility at the Troubleshooting ImportError section at Troubleshooting — NumPy v2.1.dev0 Manual for indications on how to solve this problem . (Triggered internally at /home/nvidia/work/pytorch/torch/csrc/utils/tensor_numpy.cpp:68.)
_dtype_to_storage = {data_type(0).dtype: data_type for data_type in _storages}
Traceback (most recent call last):
File “infer-det.py”, line 86, in
main(args)
File “infer-det.py”, line 34, in main
dwdh = torch.asarray(dwdh * 2, dtype=torch.float32, device=device)
AttributeError: module ‘torch’ has no attribute ‘asarray’
root@tx2:/home/test/YOLOv8-TensorRT#

How can I fix it?

There is no update from you for a period, assuming this is not an issue any more.
Hence we are closing this topic. If need further support, please open a new one.
Thanks

Could you check what PyTorch version is required by the YOLOv8 sample first?
Thanks.