Skip to content

Commit

Permalink
add onnx-rt from conda,
Browse files Browse the repository at this point in the history
add empty line,
rm unnecessary file.

Signed-off-by: jafraustro <[email protected]>
  • Loading branch information
jafraustro committed Nov 27, 2024
1 parent 4bba2a2 commit ac21a35
Show file tree
Hide file tree
Showing 3 changed files with 2 additions and 14 deletions.
1 change: 1 addition & 0 deletions preset/classical-ml/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,2 +1,3 @@
jupyterhub==5.2.1
jupyter-server-proxy==4.4.0
dataset-librarian==1.0.4
10 changes: 1 addition & 9 deletions preset/inference-optimization/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -28,15 +28,7 @@ RUN ENVS_LIST=('pytorch-cpu' 'pytorch-gpu' 'tensorflow-cpu' 'tensorflow-gpu') &&
conda install -yn "$CONDA_ENV" \
"neural-compressor=${NEURAL_COMPRESSOR_VERSION}" \
'scikit-learn>=1.5.0' \
; \
done && \
conda clean -y --all

RUN ENVS_LIST=('tensorflow-cpu' 'tensorflow-gpu') && \
for i in "${!ENVS_LIST[@]}"; do \
CONDA_ENV="${ENVS_LIST[i]}" && \
conda run -n "$CONDA_ENV" python -m pip install --no-deps --no-cache-dir \
'onnxruntime>=1.18.1' \
'onnxruntime>=1.18.1'
; \
done && \
conda clean -y --all
Expand Down
5 changes: 0 additions & 5 deletions preset/inference-optimization/requirements.txt

This file was deleted.

0 comments on commit ac21a35

Please sign in to comment.