Skip to content

Commit 399652e

Browse files
vishalbolludeliahu
authored andcommitted
Improve system package documentation (#851)
(cherry picked from commit 25e59e2)
1 parent f978586 commit 399652e

File tree

1 file changed

+15
-4
lines changed

1 file changed

+15
-4
lines changed

docs/dependency-management/system-packages.md

Lines changed: 15 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
Cortex uses Docker images to deploy your models. These images can be replaced with custom images that you can augment with your system packages and libraries. You will need to push your custom images to a container registry that your cluster has access to (e.g. [Docker Hub](https://hub.docker.com) or [AWS ECR](https://aws.amazon.com/ecr)).
44

5-
See the `image paths` section in [cluster configuration](../cluster-management/config.md) for a complete list of customizable images.
65

76
## Create a custom image
87

@@ -12,12 +11,24 @@ Create a Dockerfile to build your custom image:
1211
mkdir my-api && cd my-api && touch Dockerfile
1312
```
1413

15-
Specify the base image you want to override followed by your customizations. The sample Dockerfile below inherits from Cortex's Python serving image and installs the `tree` system package.
14+
The Docker images used to deploy your models are listed below. Based on the Cortex Predictor and compute type specified in your API configuration, choose a Cortex image to use as the base for your custom Docker image.
1615

16+
<!-- CORTEX_VERSION_README x5 -->
17+
* Python (CPU): cortexlabs/python-serve:0.14.0
18+
* Python (GPU): cortexlabs/python-serve-gpu:0.14.0
19+
* TensorFlow (CPU or GPU): cortexlabs/tf-api:0.14.0
20+
* ONNX (CPU): cortexlabs/onnx-serve:0.14.0
21+
* ONNX (GPU): cortexlabs/onnx-serve-gpu:0.14.0
22+
23+
Note that the Docker image version must match your cluster version displayed in `cortex version`.
24+
25+
The sample Dockerfile below inherits from Cortex's Python CPU serving image and installs the `tree` system package.
26+
27+
<!-- CORTEX_VERSION_README -->
1728
```dockerfile
1829
# Dockerfile
1930

20-
FROM cortexlabs/python-serve
31+
FROM cortexlabs/python-serve:0.14.0
2132

2233
RUN apt-get update \
2334
&& apt-get install -y tree \
@@ -68,7 +79,7 @@ cortex cluster update --config=cluster.yaml
6879

6980
## Use system packages in workloads
7081

71-
Cortex will use your image to launch Python serving workloads and you will have access to any packages you added:
82+
Cortex will use your custom image to launch workloads and you will have access to any packages you added:
7283

7384
```python
7485
# predictor.py

0 commit comments

Comments
 (0)