File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -65,8 +65,10 @@ RUN if [ "$MACHINE_TYPE" = "gpu" ]; then export USE_CUDA=1; fi \
65
65
&& git clone https://github.com/pytorch/serve.git \
66
66
&& cd serve \
67
67
&& git checkout ${BRANCH_NAME} \
68
- && python$PYTHON_VERSION -m venv /home/venv \
69
- && python -m pip install -U pip setuptools \
68
+ && python$PYTHON_VERSION -m venv /home/venv
69
+ ENV PATH="/home/venv/bin:$PATH"
70
+ WORKDIR serve
71
+ RUN python -m pip install -U pip setuptools \
70
72
&& if [ -z "$CUDA_VERSION" ]; then python ts_scripts/install_dependencies.py --environment=dev; else python ts_scripts/install_dependencies.py --environment=dev --cuda $CUDA_VERSION; fi \
71
73
&& if [ "$BUILD_WITH_IPEX" = "true" ]; then python -m pip install --no-cache-dir intel_extension_for_pytorch==${IPEX_VERSION} -f ${IPEX_URL}; fi \
72
74
&& python ts_scripts/install_from_src.py \
You can’t perform that action at this time.
0 commit comments