[Meta Schedule] Error when tuning a whole net with tensorcore

Troubleshooting
#1

I imitate this test file https://github.com/apache/tvm/blob/main/tests/python/integration/test_auto_tensorize.py, and I just want to use meta schedule to optimize a whole net using tensorcore.

But I met this problem for days. Here is the error message:

屏幕截图 2023-04-12 181544
屏幕截图 2023-04-12 1815441917×1160 180 KB
My source code is on github

Can anyone help me solve the error

Thanks

#2

I solved the problem by removing pytorchcu116 and reinstall a CPU version.

I googled the problem and it seems that cuda11.6 has some trouble with pytorch cu116 and TVM