Multi-Task-Transformer icon indicating copy to clipboard operation
Multi-Task-Transformer copied to clipboard

Issue running inference.py - proper requirements file needed

Open op1009 opened this issue 1 year ago • 0 comments

  • [X] Downloaded pretrained model for cityscapes-3d

  • [X] installed requirements as mentioned here https://github.com/prismformore/Multi-Task-Transformer/pull/23

  • while running inference script

!CUDA_VISIBLE_DEVICES=0 python inference.py --config_path='./configs/cityscapes3d/cs_swinB_taskprompter.yml' --image_path='/content/IMG_20200209_2301.jpg' --ckp_path='/content/drive/MyDrive/YOLOv8/TaskPrompter_CS_swinB_v2.pth.tar' --save_dir='/content/'

error occurs as

File "/content/Multi-Task-Transformer/TaskPrompter/detection_toolbox/det_losses.py", line 7, in <module>
    import mmcv
ModuleNotFoundError: No module named 'mmcv' 
  • As an expected solution, tried pip install mmcv, but it takes indefinite time(waited >20minutes) Building wheels for collected packages: mmcv .
  • Also tried some solutions from Installation takes forever - mmcv issue but still no solution worked

System Specifications

  • Google Colab T4 GPU
  • Python 3.10.12
  • Cuda
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:33:58_PDT_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0

P.S. - As per README this is built using Python 3.7 but had no idea about other library versions so continued without specifying any

@prismformore

op1009 avatar Dec 02 '23 20:12 op1009