ml-agents icon indicating copy to clipboard operation
ml-agents copied to clipboard

Published 20 hours ago • verified publisher icon Unity-Technologies

Support for Torch 1.9

Open amogkam opened this issue 4 years ago • 4 comments

Is support for PyTorch 1.9 on the roadmap for mlagents? Currently torch is constrained to be less than 1.9 in setup.py.

amogkam avatar Sep 14 '21 00:09 amogkam

Seconded, especially as the installation instructions for local development have you install the latest stable PyTorch, which is 1.10. Also, I am on an M1 mac and would like to run native ARM code, and a native ARM build of PyTorch is only available for 1.10 or later.

TV4Fun avatar Oct 22 '21 01:10 TV4Fun

Supporting M1 mac is definitely on our roadmap! I've logged the upgrade to PyTorch in our internal tracker as MLA-2273 and will be prioritized accordingly.

maryamhonari avatar Oct 22 '21 22:10 maryamhonari

I was able to run this with Torch 1.10 just by updating the version requirement in the setup.py file, and it seems to be working fine.

TV4Fun avatar Oct 22 '21 23:10 TV4Fun

agree that changing the requirements to "torch>=1.8.0,<=1.10.2;(platform_system!='Windows' and python_version>='3.9')"

it works well on m1, too!

StefanoCecere avatar Jan 29 '22 01:01 StefanoCecere