XLA.jl - Compiling Julia to XLA
NOTE: We're in the process of adding better instructions. Check back in a bit.
Getting started on TPUs
Running on Colab
Google currently offers free access to Cloud TPUs through its Colab notebook
service. Colab does not officially support julia at the moment, but it is
possible to install julia by manually installing it into the runtime (though
this 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 installation
notebook in docs/colab/InstallJuliaXLA.ipynb:
Afterwards, any JuliaTPU notebook should work if opened without resetting the runtime in between.
Running on GCP
The process for setting up this repository to run against TPUs is much the same as the process for setting up the repository locally. However, since there is additional steps involved in launching the actual TPU, we are providing a tutorial to walk you through all the steps. It is recommended for those new to Julia and/or TPUs. If you're already familiar with both, you may skip the tutorial and just use the setup guide below. The tutorial will open in Google Cloud Shell, by clicking the button below:
Getting started (CPU/GPU backend)
- Grab julia on branch kf/tpu3 (Prebuilt Linux x86_64 binaries with TPU support are available here)
- Instantiate this repo
julia> using TensorFlow- Get yourself an
xrt_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 in
examples/vgg_forward.jl