Simon, I opened an issue on your TIL repo with the pip incantation that I think will get the GPU working.

https://github.com/simonw/til/issues/69

I ran into that previously