search

shenyunhang / APE

[CVPR 2024] Aligning and Prompting Everything All at Once for Universal Visual Perception
https://arxiv.org/abs/2312.02153
Apache License 2.0
459 stars 28 forks source link

running env. #8

Closed likyoo closed 7 months ago

likyoo commented 7 months ago

Thanks for your excellent work. Could you provide versions of the relevant dependencies, such as the CUDA. Or a Docker image?

I'm getting many errors when installing and compiling detectron and detrex ....

Thanks a lot ~

shenyunhang commented 7 months ago

The cuda env is cuda 11.3. so I install the main package via

pip3 install torch==1.12.1 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 --upgrade
pip3 install transformers==4.32.1

And xformer and apex are complied from the source.

likyoo commented 7 months ago

Thanks for your reply :) I solved these issues by using docker ~

xijun-cs commented 3 months ago

@likyoo Hi, would you please share the docker image with me?

Best

wdc233 commented 1 month ago

And xformer and apex are complied from the source 从源码编译，他不torch也得>2.2版本，你1.2得torch是如何安装成功的呢，这个过程能否分享一下