Xilinx / Vitis-AI-Copyleft-Model-Zoo

8 stars 2 forks source link

Errors in replicating yolov7 build - what is the expected docker environment setup #9

Open rcocorad opened 3 months ago

rcocorad commented 3 months ago

I'm trying to build a yolov7 xmodel following the Readme and the jyupter noteback tutorial. I immediately get a string of errors then get stuck. I'm using docker vitis ait 3.5 gpu: Docker Image Version: 3.5.0.001-8108149 (GPU) Vitis AI Git Hash: 8108149 Build Date: 2024-02-26 WorkFlow: pytorch


ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This pytorch-nndct 3.5.0 requires ninja, which is not installed. pytorch-nndct 3.5.0 requires scipy<=1.9.3, but you have scipy 1.10.1 which is incompatible.


rcocorad commented 3 months ago

Update: I was able to get the train/eval scripts to run by creating a new conda environment from python 3.8, pip installing the requirements.txt, updating torch to 2.2.1+cu121, torchvision to 0.17.1+cu121, installing xir manually via the conda 3.5.0 channel, and isntalling pytorch nndct from script in the vitis-ai-3.5.0 repo.

It would be nice to get a docker image or conda environment that works out of the box.