we got the wheels from:
https://download.pytorch.org/whl/torch_stable.html 16
we also build from source, but that ended up being 2.4GB.  Not sure if there are better compiler flags / env variables to make it smaller.