[Feature Request] Publish wheels for OS X and ARM64

[Sitin]

It seems that the torchrl can run at arm64 and Apple Silicone. It would be nice to have the same requirements.txt/Pipfile at all platforms where this library can be executed.

[vmoens]

cc @osalpekar
How should we proceed to have this?

[osalpekar]

@Sitin @vmoens Wanted to clarify the ask here - IIUC all the wheels build requirements are explicitly listed here: https://github.com/pytorch/rl/blob/main/setup.py#L208, and are unified across platforms. Is the ask for official binaries for arm64/Apple Silicon?

[hyerra]

I am also running into this issue, and not sure if this is the same, but essentially when I try to install torchrl on my Apple Silicon Mac, I run into this:

pip install torchrl
ERROR: Could not find a version that satisfies the requirement torchrl (from versions: none)
ERROR: No matching distribution found for torchrl

I have to pip install using the url directly using this command:

pip install https://github.com/pytorch/rl.git

pip install torchrl works on other machines though like Windows. I'm not sure if this thread is relevant? It looks like they added the universal packages like 2 weeks ago and the last version of torchrl was before that so maybe publishing a new version would fix?

[Sitin]

@osalpekar, just want to clarify. As @hyerra mentioned, while it is possible to install pytorchlr on M1/M2 from the repository but I can't do the same using package version (i.e. pip install torchrl).

When it comes to the implementation, then platform-specific wheels with prebuilt binnaries would be my favorite options (and probably the easiest for you since CircleCI has M1 executors).

[osalpekar]

Thanks for clarifying. The issue is essentially that we don't publish first-party wheels for Apple Silicon or Arm64, but adding support is straightforward (at least for Apple Silicon).

cc @atalman We should add Nova pipelines for TorchRL for M1, as well as for aarch64 once those are ready.

The reason the pip install works out of the box for Windows is that we are publishing wheels for Linux/Windows/Mac x86. The github link works since that seems to just check out the repo and build the wheel from source locally.

[Sitin]

@osalpekar, than you very much! Can't wait to finally fix my dependecy mess.

[vmoens]

v0.2.0 now includes wheels for M1!
Feel free to reopen if issues persist