-
Notifications
You must be signed in to change notification settings - Fork 95
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Update PyTorch version in install_nvidia.sh #439
Conversation
Hello Stefania, I am gonna have a look into that and update to the closest version available soon |
Hi @Pavlos-Tosidis, did you have a chance to look into this? |
Yeah, meant to comment on it on todays TMT meeting. |
In that case, I believe we should drop support for TX2 in the main branch of this repository otherwise we will be blocked with updating PyTorch. Using willing to use the unsupported TX2 may still revert to an older commit on this repo or older images (which we should document). |
One other possibility would be to keep the installation for tx-2 only for the tools not failing, i.e., keep the script as it is and add a notice with a list of tools that do not work in this platform (unless nothing works in the updated version, so this should be indeed tackled in a different manner). |
We can keep the current version on TX-2 image, so we can close this for now. |
In #431 we updated the module versions but this script was not updated:
In this PR, I propose some changes to update the Python modules version but I'm not sure if they will work on the NVIDIA TX2/AGX/NX .