File tree 2 files changed +14
-15
lines changed
2 files changed +14
-15
lines changed Original file line number Diff line number Diff line change @@ -315,16 +315,15 @@ ARG XPU_SMI_VERSION
315
315
316
316
ARG API_URL=https://api.github.com/repos/intel/xpumanager/releases/tags/V${XPU_SMI_VERSION}
317
317
318
- RUN curl -H "Accept: application/vnd.github.v3+json" -H "User-Agent: MyClient/1.0.0" "$API_URL" >> /tmp/asset_data.txt && \
319
- wget -q --no-check-certificate $(cat /tmp/asset_data.txt | jq -r '.assets[] | select(.name | test("^xpu-smi.*u22\\ .04_amd64\\ .deb$")) | .browser_download_url' ) && \
320
- ldconfig && dpkg -i --force-all *.deb && \
321
- rm -rf *.deb /etc/apt/sources.list.d/intel-gpu-jammy.list /etc/apt/sources.list.d/oneAPI.list /tmp/asset_data.txt
318
+ RUN curl -H "Accept: application/vnd.github.v3+json" -H "User-Agent: MyClient/1.0.0" "$API_URL" | tee /tmp/asset_data.txt && \
319
+ curl -L $(cat /tmp/asset_data.txt | jq -r '.assets[] | select(.name | test("^xpu-smi.*u22\\ .04_amd64\\ .deb$")) | .browser_download_url' ) | xargs -I {} dpkg -i --force-all {} && \
320
+ ldconfig && \
321
+ rm -rf -- /etc/apt/sources.list.d/intel-gpu-jammy.list /etc/apt/sources.list.d/oneAPI.list /tmp/asset_data.txt
322
322
323
323
ARG GID=109
324
324
325
- RUN groupadd -g ${GID} render
326
-
327
- RUN usermod -aG video,render model-server
325
+ RUN groupadd -g ${GID} render &&\
326
+ usermod -aG video,render model-server
328
327
329
328
USER model-server
330
329
Original file line number Diff line number Diff line change 27
27
# https://docs.docker.com/develop/develop-images/build_enhancements/
28
28
29
29
if [[ " $1 " == " cpu" ]]; then
30
- wget https://download.pytorch.org/models/squeezenet1_1-b8a52dc0.pth
31
- torch-model-archiver --model-name squeezenet1_1 --version 1.1 --model-file /home/model-server/model-archive/model.py --serialized-file squeezenet1_1-b8a52dc0.pth --handler image_classifier --export-path /home/model-server/model-store
32
- rm -rf squeezenet1_1-b8a52dc0.pth
30
+ wget https://download.pytorch.org/models/squeezenet1_1-b8a52dc0.pth
31
+ torch-model-archiver --model-name squeezenet1_1 --version 1.1 --model-file /home/model-server/model-archive/model.py --serialized-file squeezenet1_1-b8a52dc0.pth --handler image_classifier --export-path /home/model-server/model-store
32
+ rm -rf squeezenet1_1-b8a52dc0.pth
33
33
elif [[ " $1 " == " xpu" ]]; then
34
- python /home/model-server/model-archive/ipex_squeezenet.py
35
- torch-model-archiver --model-name squeezenet1_1 --version 1.1 --serialized-file squeezenet1_1-jit.pt --handler image_classifier --export-path /home/model-server/model-store
36
- rm -rf squeezenet1_1-jit.pt
34
+ python /home/model-server/model-archive/ipex_squeezenet.py
35
+ torch-model-archiver --model-name squeezenet1_1 --version 1.1 --serialized-file squeezenet1_1-jit.pt --handler image_classifier --export-path /home/model-server/model-store
36
+ rm -rf squeezenet1_1-jit.pt
37
37
else
38
- echo " Only cpu and xpu devices supported"
39
- exit 1
38
+ echo " Only cpu and xpu devices supported"
39
+ exit 1
40
40
fi
41
41
42
42
[ -f " /home/model-server/model-store/squeezenet1_1.mar" ] && echo " squeezenet1_1.pth Archived Succesfully at /home/model-server/model-store/squeezenet1_1.mar"
You can’t perform that action at this time.
0 commit comments