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

$ docker run --runtime=nvidia -it --rm -p 8888:8888 refikanadolstudio/colab-local:latest

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

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

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

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)