Multi-Task-Transformer
Multi-Task-Transformer copied to clipboard
Issue running inference.py - proper requirements file needed
-
[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