Poetry can't install torch nightly

vwxyzjn / cleanrl

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

http://docs.cleanrl.dev

Other

5.54k stars 631 forks source link

Poetry can't install torch nightly #247

Closed nyp0x closed 2 years ago

nyp0x commented 2 years ago

I tried to install the latest torch v1.13 with m1 mac gpu support, torch = {file = "/path/to/torch-1.13.0.dev20220802-cp38-none-macosx_11_0_arm64.whl"} but it seems poetry can't handle it properly:

  AttributeError
  'NoneType' object has no attribute 'satisfies'

vwxyzjn commented 2 years ago

Thanks for reporting. You can always do poetry run pip install …wheel. This week I don’t have my laptop but I can take a further look next week.

nyp0x commented 2 years ago

Thank you, it works with torch pip-installed on top of poetry. It seems many ops are not yet implemented for mps device though. E.g, can't sample a distribution:

samples_2d = torch.multinomial(probs_2d, sample_shape.numel(), True).T
NotImplementedError: The operator 'aten::multinomial' is not current implemented for the MPS device