pytorch 模块'torch.onnx.symbolic_helper'没有属性'quantized_args'

agxfikkp  于 2022-12-13  发布在  其他
关注(0)|答案(1)|浏览(419)

当从torchvision导入转换时,我得到以下错误

module 'torch.onnx.symbolic_helper' has no attribute 'quantized_args'

我执行导入

from torchvision import transforms

在出现错误之前,它会抛出以下警告:

OnnxExporterWarning: Symbolic function 'aten::_shape_as_tensor' already registered for opset 9. Replacing the existing function with new function. This is unexpected. Please report it on https://github.com/pytorch/pytorch/issues.
      errors.OnnxExporterWarning,

Torch 和 Torch 版本

torch==1.13.0
torchvision==0.14.0

其他信息:我有cuda启用gpu使用cuda 11.8。

sczxawaw

sczxawaw1#

在我的情况下,这是版本冲突错误,尝试改变 Torch ,只有cpu版本,应该工作,如果你想坚持使用gpu尝试使用cuda==11.6沿着 Torch ==1.13.0和 Torch 视觉==0.14.0。

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

请尝试引用此链接。

https://pytorch.org/get-started/locally/

相关问题