import error

[wm901115nwpu]

🐛 Describe the bug

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/Users/unicorn/miniconda3/envs/pt2/lib/python3.10/site-packages/torchvision/__init__.py", line 6, in <module>
    from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils
  File "/Users/unicorn/miniconda3/envs/pt2/lib/python3.10/site-packages/torchvision/_meta_registrations.py", line 100, in <module>
    def meta_nms(dets, scores, iou_threshold):
  File "/Users/unicorn/miniconda3/envs/pt2/lib/python3.10/site-packages/torch/library.py", line 417, in inner
    handle = entry.abstract_impl.register(func, source)
  File "/Users/unicorn/miniconda3/envs/pt2/lib/python3.10/site-packages/torch/_library/abstract_impl.py", line 30, in register
    if torch._C._dispatch_has_kernel_for_dispatch_key(self.qualname, "Meta"):
RuntimeError: operator torchvision::nms does not exist

Versions

Collecting environment information...
PyTorch version: 2.2.0a0+git2d5fec4
Is debug build: True
CUDA used to build PyTorch: None
ROCM used to build PyTorch: N/A

OS: macOS 14.1 (arm64)
GCC version: Could not collect
Clang version: 17.0.4
CMake version: version 3.26.4
Libc version: N/A

Python version: 3.10.11 (main, May 17 2023, 14:30:36) [Clang 14.0.6 ] (64-bit runtime)
Python platform: macOS-14.1-arm64-arm-64bit
Is CUDA available: False
CUDA runtime version: No CUDA
CUDA_MODULE_LOADING set to: N/A
GPU models and configuration: No CUDA
Nvidia driver version: No CUDA
cuDNN version: No CUDA
HIP runtime version: N/A
MIOpen runtime version: N/A
Is XNNPACK available: True

CPU:
Apple M2 Pro

Versions of relevant libraries:
[pip3] iree-torch==0.0.1
[pip3] memory-efficient-attention-pytorch==0.1.6
[pip3] numpy==1.24.3
[pip3] onnx==1.16.0
[pip3] onnx-graphsurgeon==0.3.26
[pip3] onnx-tool==0.6.4
[pip3] onnxconverter-common==1.14.0
[pip3] onnxoptimizer==0.3.17
[pip3] onnxruntime-extensions==0.10.0+2cf9bab
[pip3] onnxruntime-silicon==1.16.0
[pip3] onnxscript==0.1.0.dev20231021
[pip3] onnxsim==0.4.33
[pip3] skl2onnx==1.15.0
[pip3] torch==2.2.0a0+git98564d2
[pip3] torch-mlir==0.0.1
[pip3] torchaudio==2.2.0a0+6e26515
[pip3] torchdata==0.7.0a0+ce07539
[pip3] torchinfo==1.8.0
[pip3] torchvision==0.17.0a0+209b2b3
[conda] iree-torch 0.0.1 pypi_0 pypi
[conda] memory-efficient-attention-pytorch 0.1.6 pypi_0 pypi
[conda] numpy 1.24.3 pypi_0 pypi
[conda] torch 2.2.0a0+git98564d2 pypi_0 pypi
[conda] torch-mlir 0.0.1 pypi_0 pypi
[conda] torchaudio 2.2.0a0+6e26515 pypi_0 pypi
[conda] torchdata 0.7.0a0+ce07539 pypi_0 pypi
[conda] torchinfo 1.8.0 pypi_0 pypi
[conda] torchvision 0.17.0a0+209b2b3 pypi_0 pypi

[NicolasHug]

Hi @wm901115nwpu can you please share the code snippet you tried to run?

[wm901115nwpu]

╭─    ~/workspace ──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────── ✔  pt2   08:19:13 上午  ─╮
╰─ python ─╯
Python 3.10.11 (main, May 17 2023, 14:30:36) [Clang 14.0.6 ] on darwin
Type "help", "copyright", "credits" or "license" for more information.

import torchvision
Traceback (most recent call last):
File "", line 1, in
File "/Users/unicorn/miniconda3/envs/pt2/lib/python3.10/site-packages/torchvision/init.py", line 6, in
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils
File "/Users/unicorn/miniconda3/envs/pt2/lib/python3.10/site-packages/torchvision/_meta_registrations.py", line 100, in
def meta_nms(dets, scores, iou_threshold):
File "/Users/unicorn/miniconda3/envs/pt2/lib/python3.10/site-packages/torch/library.py", line 417, in inner
handle = entry.abstract_impl.register(func, source)
File "/Users/unicorn/miniconda3/envs/pt2/lib/python3.10/site-packages/torch/_library/abstract_impl.py", line 30, in register
if torch._C._dispatch_has_kernel_for_dispatch_key(self.qualname, "Meta"):
RuntimeError: operator torchvision::nms does not exist

[NicolasHug]

Do import torch first?

[wm901115nwpu]

─    ~/workspace ───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────── ✔  8m 37s   pt2   08:27:53 上午  ─╮
╰─ python ─╯
Python 3.10.11 (main, May 17 2023, 14:30:36) [Clang 14.0.6 ] on darwin
Type "help", "copyright", "credits" or "license" for more information.

import torch
import torchvision
Traceback (most recent call last):
File "", line 1, in
File "/Users/unicorn/miniconda3/envs/pt2/lib/python3.10/site-packages/torchvision/init.py", line 6, in
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils
File "/Users/unicorn/miniconda3/envs/pt2/lib/python3.10/site-packages/torchvision/_meta_registrations.py", line 100, in
def meta_nms(dets, scores, iou_threshold):
File "/Users/unicorn/miniconda3/envs/pt2/lib/python3.10/site-packages/torch/library.py", line 417, in inner
handle = entry.abstract_impl.register(func, source)
File "/Users/unicorn/miniconda3/envs/pt2/lib/python3.10/site-packages/torch/_library/abstract_impl.py", line 30, in register
if torch._C._dispatch_has_kernel_for_dispatch_key(self.qualname, "Meta"):
RuntimeError: operator torchvision::nms does not exist

[wm901115nwpu]

same error

[NicolasHug]

Did you try to build form source? Or install the nightlies

[wm901115nwpu]

the error is that I use pytorch and torchvision building from source, I use nightly version import is OK.I guess pytorch the source is something wrong.

[NicolasHug]

Hm yeah, it looks like something is going wrong when you build pytorch from source then

[wm901115nwpu]

I try to delete build folder, rebuild it

[wm901115nwpu]

I update torchvision and torch from source, rebuild, import is OK。

[huydhn]

I'm seeing the same issue on my end building from latest main from both torch and torchvision. I suspect that it has something to do with this change pytorch/pytorch#113182. Nightly CPU build works fine.