Error when finding Inference on VM

[harshitha-batta]

Hello,
I am new to detectron2, I did run my code on a colab notebook and it works fine, but now when I am running my code on a Ubuntu 20.04 VM on Microsoft Azure with Tesla K80 GPU, I am getting an attribute error as shown in the image.

I do have detectron2 , cuda , pytorch and tensorflow installed. I even checked if tensorflow and pytorch are using the GPU,and I am not sure why I am facing this issue.
This is my code for loading the model for inference. The output folder with the weights along and the pickle file are in the current directory

If anyone knows why I could be getting this error, please let me know.