search

aws-neuron / aws-neuron-sdk

Powering AWS purpose-built machine learning chips. Blazing fast and cost effective, natively integrated into PyTorch and TensorFlow and integrated with your favorite AWS services
https://aws.amazon.com/machine-learning/neuron/
Other
462 stars 154 forks source link

torch-neuronx pip package has erroneous requirements #834

Open marty-sullivan opened 9 months ago

marty-sullivan commented 9 months ago

Not sure where the build code for the torch-neuronx>=2 python package is. If you point me to it, I'd be happy to make a PR.

Problem: it seems that there is an erroneous version requirement listed, causing issues with my build tools (below output from poetry):

Invalid requirement (libneuronxla (<3.0,>2.0.x)) found in torch-neuronx-2.1.1.2.0.1b0 dependencies, skipping

My workaround is to explicitly require libneuronxla^2.0, however, fixing the requirement string on your end to <3.0,>=2.0 should solve the problem.

aws-rhsoln commented 9 months ago

Thank you for creating an issue. We are looking into this.

aws-rhsoln commented 9 months ago

We have added a fix for the requirements and should be available in the upcoming release