|
225 | 225 | "outputs": [],
|
226 | 226 | "source": [
|
227 | 227 | "!pip3 install {USER_FLAG} google-cloud-aiplatform --upgrade\n",
|
228 |
| - "!pip3 install {USER_FLAG} kfp google-cloud-pipeline-components --upgrade" |
| 228 | + "!pip3 install {USER_FLAG} google-cloud-pipeline-components==0.1.7 kfp --upgrade" |
229 | 229 | ]
|
230 | 230 | },
|
231 | 231 | {
|
|
1300 | 1300 | " project\n",
|
1301 | 1301 | " )\n",
|
1302 | 1302 | "\n",
|
| 1303 | + " gcc_aip.ModelUploadOp.component_spec.implementation.container.image = \"gcr.io/ml-pipeline/google-cloud-pipeline-components:0.1.7\"\n", |
1303 | 1304 | " model_upload_op = gcc_aip.ModelUploadOp(\n",
|
1304 | 1305 | " project=project,\n",
|
1305 | 1306 | " display_name=model_display_name,\n",
|
|
1309 | 1310 | " serving_container_ports=[PORT] \n",
|
1310 | 1311 | " )\n",
|
1311 | 1312 | " \n",
|
| 1313 | + " gcc_aip.EndpointCreateOp.component_spec.implementation.container.image = \"gcr.io/ml-pipeline/google-cloud-pipeline-components:0.1.7\"\n", |
1312 | 1314 | " endpoint_create_op = gcc_aip.EndpointCreateOp(\n",
|
1313 | 1315 | " project=project,\n",
|
1314 | 1316 | " display_name=model_display_name,\n",
|
1315 | 1317 | " )\n",
|
1316 |
| - "\n", |
| 1318 | + " gcc_aip.ModelDeployOp.component_spec.implementation.container.image = \"gcr.io/ml-pipeline/google-cloud-pipeline-components:0.1.7\"\n", |
1317 | 1319 | " model_deploy_op = gcc_aip.ModelDeployOp(\n",
|
1318 | 1320 | " project=project,\n",
|
1319 | 1321 | " endpoint=endpoint_create_op.outputs[\"endpoint\"],\n",
|
|
1726 | 1728 | " project\n",
|
1727 | 1729 | " )\n",
|
1728 | 1730 | "\n",
|
| 1731 | + " gcc_aip.ModelUploadOp.component_spec.implementation.container.image = \"gcr.io/ml-pipeline/google-cloud-pipeline-components:0.1.7\"\n", |
1729 | 1732 | " model_upload_op = gcc_aip.ModelUploadOp(\n",
|
1730 | 1733 | " project=project,\n",
|
1731 | 1734 | " display_name=model_display_name,\n",
|
1732 | 1735 | " serving_container_image_uri=build_image_task.outputs['serving_container_uri'],\n",
|
1733 | 1736 | " serving_container_predict_route=\"/predictions/{}\".format(MAR_MODEL_NAME),\n",
|
1734 | 1737 | " serving_container_health_route=\"/ping\",\n",
|
1735 |
| - " serving_container_ports=[PORT] \n", |
| 1738 | + " serving_container_ports=[PORT]\n", |
1736 | 1739 | " )\n",
|
1737 |
| - " \n", |
| 1740 | + "\n", |
| 1741 | + " gcc_aip.EndpointCreateOp.component_spec.implementation.container.image = \"gcr.io/ml-pipeline/google-cloud-pipeline-components:0.1.7\"\n", |
1738 | 1742 | " endpoint_create_op = gcc_aip.EndpointCreateOp(\n",
|
1739 | 1743 | " project=project,\n",
|
1740 | 1744 | " display_name=model_display_name,\n",
|
1741 | 1745 | " )\n",
|
1742 | 1746 | "\n",
|
| 1747 | + " gcc_aip.ModelDeployOp.component_spec.implementation.container.image = \"gcr.io/ml-pipeline/google-cloud-pipeline-components:0.1.7\"\n", |
1743 | 1748 | " model_deploy_op = gcc_aip.ModelDeployOp(\n",
|
1744 | 1749 | " project=project,\n",
|
1745 | 1750 | " endpoint=endpoint_create_op.outputs[\"endpoint\"],\n",
|
1746 | 1751 | " model=model_upload_op.outputs[\"model\"],\n",
|
1747 | 1752 | " deployed_model_display_name=model_display_name,\n",
|
1748 | 1753 | " machine_type=\"n1-standard-4\",\n",
|
1749 | 1754 | " accelerator_type='NVIDIA_TESLA_P100',\n",
|
1750 |
| - " accelerator_count=1 \n", |
| 1755 | + " accelerator_count=1\n", |
1751 | 1756 | " )\n",
|
1752 | 1757 | "\n"
|
1753 | 1758 | ]
|
|
1958 | 1963 | "provenance": [],
|
1959 | 1964 | "toc_visible": true
|
1960 | 1965 | },
|
| 1966 | + "environment": { |
| 1967 | + "kernel": "python3", |
| 1968 | + "name": "tf2-gpu.2-6.m82", |
| 1969 | + "type": "gcloud", |
| 1970 | + "uri": "gcr.io/deeplearning-platform-release/tf2-gpu.2-6:m82" |
| 1971 | + }, |
1961 | 1972 | "kernelspec": {
|
1962 | 1973 | "display_name": "Python 3",
|
1963 | 1974 | "language": "python",
|
|
1973 | 1984 | "name": "python",
|
1974 | 1985 | "nbconvert_exporter": "python",
|
1975 | 1986 | "pygments_lexer": "ipython3",
|
1976 |
| - "version": "3.8.2" |
| 1987 | + "version": "3.7.10" |
1977 | 1988 | }
|
1978 | 1989 | },
|
1979 | 1990 | "nbformat": 4,
|
|
0 commit comments