Repository files navigation

NOTE: We're in the process of adding better instructions. Check back in a bit.

Google currently offers free access to Cloud TPUs through its Colab notebookservice. Colab does not officially support julia at the moment, but it ispossible to install julia by manually installing it into the runtime (thoughthis has to be done every time the runtime gets reset). By this mechanism,you can get access to TPUs through the notebook interface. Start with the installationnotebook indocs/colab/InstallJuliaXLA.ipynb:

Run in Google Colab

Afterwards, any JuliaTPU notebook should work if opened without resetting the runtimein between.

The process for setting up this repository to run against TPUs is muchthe same as the process for setting up the repository locally. However, sincethere is additional steps involved in launching the actual TPU, we are providinga tutorial to walk you through all the steps. It is recommended for those new toJulia and/or TPUs. If you're already familiar with both, you may skip the tutorialand just use the setup guide below. The tutorial will open in Google Cloud Shell,by clicking the button below:

• Grab julia on branchkf/tpu3 (Prebuilt Linux x86_64 binaries with TPU support are availablehere)
• Instantiate this repo
• julia> using TensorFlow
• Get yourself anxrt_server (either running it locally viarun(`$(joinpath(dirname(pathof(TensorFlow)),"..","deps","downloads","bin","xrt_server"))`) or by spinning up a Google Cloud TPU and starting an SSH tunnel to expose its port to the world) and connect it to localhost:8470
• Run the script inexamples/vgg_forward.jl