Closed lilpenguin closed 3 years ago
This problem Unrecognized attribute: split for operator Split
might be solved by downgrading your onnx version to 1.6.0
Downgrading onnx to 1.6.0 worked, thanks! I also switched to the provided docker container, and the cuda error disappeared as well.
Hi, I installed TASO from source with cuda 10.1 and cudnn 7.6.5. The installation was inside an anaconda environment with onnx-1.8.0, python-3.6.12 and protobuf-3.14.0 installed. I was able to run resnet50 and bert under /examples. But for nasrnn, I got the following error:
In addition, running resnext50 seemed to finish correctly, but after the new graph was outputted, an onnx error appeared at the end (please see the full output here):
I ran the examples on a Tesla P100 and a Tesla V100, and in both cases, these errors happened.
Can you please help me solve these issues? Thank you for your help!