Commit a3bbfe4
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -64,6 +64,9 @@ RUN pip install papermill jupyterlab python-lsp-server[all] jupyterlab-lsp | |||
| 64 | 64 | # Additional useful packages should be added here: | |
| 65 | 65 | RUN pip install pandas | |
| 66 | 66 | ||
| 67 | + # Monkey-patch the default TPU to the local (TPU VM). | ||
| 68 | + RUN sed -i 's/def connect(tpu=None,/def connect(tpu="local",/' /usr/local/lib/python3.8/site-packages/tensorflow/python/distribute/cluster_resolver/tpu/tpu_cluster_resolver.py | ||
| 69 | + | ||
| 67 | 70 | # Set these env vars so that they don't produce errs calling the metadata server to load them: | |
| 68 | 71 | ENV TPU_ACCELERATOR_TYPE=v3-8 | |
| 69 | 72 | ENV TPU_PROCESS_ADDRESSES=local | |
0 commit comments