ugurk
June 15, 2023, 7:11am
1
I want to install Python(3.11), Pytorch(2.0.0) and Torchvision(0.15.1) with CUDA(11.8) to AGX Orin DevKit. Is it possible? Which command(s) should I use? If possible, what are the dependencies?
1 Like
@ugurk I haven’t tried this, and this is theoretically how I would go about trying it (and you may encounter issues to debug/workaround along the way). But first you would need to try building Python 3.11 (or find a PPA with aarch64 binaries to install)
Then you would need to build PyTorch from source against Python 3.11 and the version of CUDA that you have installed. You can find generic compilation instructions for PyTorch in this topic:
Below are pre-built PyTorch pip wheel installers for Jetson Nano, TX1/TX2, Xavier, and Orin with JetPack 4.2 and newer.
Download one of the PyTorch binaries from below for your version of JetPack, and see the installation instructions to run on your Jetson. These pip wheels are built for ARM aarch64 architecture, so run these commands on your Jetson (not on a host PC). You can also use the pre-built l4t-pytorch and l4t-ml container images and Dockerfiles .
PyTorch pip wheels
JetPack 5
PyTo…
Python 3.11 support in PyTorch is still relatively new, and I would recommend keeping an eye on this GitHub issue:
opened 12:01AM - 10 Oct 22 UTC
closed 11:06PM - 05 Jun 23 UTC
module: ci
triaged
enhancement
module: python frontend
Currently Python 3.11 isn't fully supported in Pytorch yet, including testing in… CI and publishing wheels to PyPI.
Python 3.11 is currently in Release Candidate state and will be released in about two weeks. Major packages like NumPy, Pandas, SciPy and Matplotlib have already uploaded their Python 3.11 wheels to PyPI.
Python 3.11 new features include:
> - [PEP 657](https://www.python.org/dev/peps/pep-0657/) -- Include Fine-Grained Error Locations in Tracebacks
> - [PEP 654](https://www.python.org/dev/peps/pep-0654/) -- Exception Groups and except*
> - [PEP 673](https://www.python.org/dev/peps/pep-0673/) -- Self Type
> - [PEP 646](https://www.python.org/dev/peps/pep-0646/)-- Variadic Generics
> - [PEP 680](https://www.python.org/dev/peps/pep-0680/)-- tomllib: Support for Parsing TOML in the Standard Library
> - [PEP 675](https://www.python.org/dev/peps/pep-0675/)-- Arbitrary Literal String Type
> - [PEP 655](https://www.python.org/dev/peps/pep-0655/)-- Marking individual TypedDict items as required or potentially-missing
> - [bpo-46752](https://bugs.python.org/issue46752)-- Introduce task groups to asyncio
> - The [Faster Cpython Project](https://github.com/faster-cpython) is already yielding some exciting results: this version of CPython 3.11 is ~ 19% faster on the geometric mean of the [PyPerformance benchmarks](https://speed.python.org/), compared to 3.10.0.
Pytorch is an import package in the deep learning stack, so support at release will help speed up Python 3.11 adoption.
It would be great to have full Python 3.11 support in Pytorch by the time Python 3.11.0 gets released, ([expected](https://peps.python.org/pep-0664/) Monday October 6th, 2022). This includes testing in CI and publishing wheels to PyPI.
This issue can be used as a tracking issue. This includes:
- [x] Pytorch builds fully on Python 3.11
- [x] All Pytorch tests pass on Python 3.11
- [x] All CI is run and green on Python 3.11
- [x] Wheels are uploaded to PyPI for at least one Pytorch release
- [x] Manylinux (x86_64)
- [x] Manylinux (aarch64)
- [x] Windows
- [x] macOS
cc @seemethere @malfet @pytorch/pytorch-dev-infra @albanD
It appears that it requires PyTorch 2.0.1 for Python 3.11. Good luck!
system
July 12, 2023, 4:50am
5
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.