File tree 1 file changed +6
-6
lines changed
1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change 1
1
FROM xilinx/vitis-ai:1.4.1.978
2
2
3
3
RUN cd /home \
4
- && git clone https://github.com/ModelTC/MQBench.git \
5
- && cd MQBench \
6
- && sed '1,2d' requirements.txt > requirements.txt
4
+ && git clone https://github.com/ModelTC/MQBench.git
7
5
8
6
RUN source /opt/vitis_ai/conda/etc/profile.d/conda.sh \
9
7
&& source /home/vitis-ai-user/.bashrc \
10
8
&& conda activate vitis-ai-pytorch \
11
9
&& cd /home/MQBench \
12
- && pip install onnx -i https://pypi.tuna.tsinghua.edu.cn/simple \
13
- && pip install .
10
+ && pip install onnx==1.8.0 \
11
+ && cp mqbench/deploy/convert_xir.py /home/
14
12
15
- RUN echo "conda activate vitis-ai-pytorch" >> /home/vitis-ai-user/.bashrc
13
+ RUN echo "conda activate vitis-ai-pytorch" >> /home/vitis-ai-user/.bashrc
14
+
15
+ RUN echo "echo usage: python /home/convert_xir.py [-h] -Q QMODEL -C CMODEL -N NAME" >> /home/vitis-ai-user/.bashrc
You can’t perform that action at this time.
0 commit comments