Added inference-cli install to parallel gpu docker build

roboflow · Feb 22, 2024 · e0d029d · e0d029d
1 parent fd6d2f8
commit e0d029d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/dockerfiles/Dockerfile.onnx.gpu.parallel b/docker/dockerfiles/Dockerfile.onnx.gpu.parallel
@@ -46,7 +46,7 @@ WORKDIR /build
 COPY . .
 RUN ln -s /usr/bin/python3 /usr/bin/python
 RUN /bin/make create_wheels_for_gpu_notebook
-RUN pip3 install dist/inference_core*.whl dist/inference_gpu*.whl dist/inference_sdk*.whl
+RUN pip3 install dist/inference_cli*.whl dist/inference_core*.whl dist/inference_gpu*.whl dist/inference_sdk*.whl
 
 WORKDIR /notebooks
 COPY examples/notebooks .