docker-colab-local

A Docker container to act as a local runtime for Google Colab.

Run

$ docker run --runtime=nvidia -it --rm -p 8081:8081 aecampos/docker-colab-local:latest

Or, to mount a volume so that it's accessible to colab:

$ docker run --runtime=nvidia -it --rm -p 8081:8081 -v /host/directory:/opt/colab aecampos/colab-local:latest

where /host/directory is a path on your host machine.

Connecting

If the container isn't running on your local machine, you'll need to forward port 8081:

$ ssh YOUR_REMOTE_MACHINE -L 8081:localhost:8081

In Colaboratory, click the "Connect" button and select "Connect to local runtime...". Enter the port 8081 step in the dialog that appears and click the "Connect" button. (from colaboratory)

Notes

• Missing some packages that come with Colab. Install them with !pip install in your notebook.