Closed bshakya77 closed 6 months ago
Hi, have you solve this problem. I met the same issue with you.
Yes, I resolved it. It was some CUDA and Pytorch version incompatibility.
In my case, the following Pytorch version worked.
Replace pytorch=1.7.1
!conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 -c pytorch
Yes, I resolved it. It was some CUDA and Pytorch version incompatibility.
Follow instructions in GitHub repo #https://github.com/fundamentalvision/Deformable-DETR
In my case, the following Pytorch version worked. Replace pytorch=1.7.1
!conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 -c pytorch
Thanks a lot. I solved it too!
Hello,
I am trying to train the FasterViT_DINO model for object detection. I installed all the required packages mentioned in the repository. But, when I run the test.py. I get the following error.
System configuration:
CUDA Version (torch): 12.1 CUDA Version (system): 12.2 Torch Version: 2.2.0+cu121 Python 3.10.13 IDE: Miniconda 3 OS: Ubuntu
I installed MultiScaleDeformableAttention 1.0 using
sh make.h
command. I think it ran sucessfully and I can see the folder MultiScaleDeformableAttention created. The log is attached below.The training works in Google Collab Pro and the configuration looks similar to my local setup.
CUDA Version (torch): 12.1 CUDA Version (system): 12.2 Torch Version: 2.1.0+cu121 Python 3.10.12
If anyone has tried object detection using Deformable DETR or FasterViT_DINO and has faced a similar issue, please provide me with the solution.
Your suggestions are greatly appreciated.
Regards, Bijay