Skip to content

Commit d3292fa

Browse files
fanyunqianTracin
fanyunqian
authored andcommitted
[Refactor] remove mqbench dependency for docker
1 parent bff9ecc commit d3292fa

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

docker/Dockerfile

+6-6
Original file line numberDiff line numberDiff line change
@@ -1,15 +1,15 @@
11
FROM xilinx/vitis-ai:1.4.1.978
22

33
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
75

86
RUN source /opt/vitis_ai/conda/etc/profile.d/conda.sh \
97
&& source /home/vitis-ai-user/.bashrc \
108
&& conda activate vitis-ai-pytorch \
119
&& 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/
1412

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

0 commit comments

Comments
 (0)