Skip to content
Build custom TensorFlow images!
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
Dockerfile
README.md
script.py

README.md

tf-dockerfile

Build custom TensorFlow images!

This repository contains a Dockerfile that contains all the dependencies needed to run a TensorFlow model on a GPU including things like Python and CUDA. The script.py contains an example script that runs a basic TensorFlow model for five epochs. Feel free to edit the script to run your own custom models.

Building and Running in Paperspace

Building this container image and running in on a GPU is easy using Paperspace. Below is a Paperspace CLI command to run this TensorFlow model. Be sure to be using the latest version of the CLI for your OS.

paperspace jobs create --useDockerfile true --buildOnly false  --command 'python /home/script.py' --apiKey XXXXXXXXXXXXXXXXXXX --workspace https://github.com/Paperspace/tf-dockerfile --machineType K80

You can also view a sample public job and even clone it in your account. Cloning will run the same CLI command included above.

Additional job options

See the docs for more options including pushing the built image to your own image repository.

You can’t perform that action at this time.