mirror of
https://github.com/pytorch/pytorch.git
synced 2025-11-11 22:34:53 +08:00
add instructions for NVIDIA Jetson platforms (#18990)
Summary: Thanks to dusty-nv , we now have Stable and Weekly wheels provided for the NVIDIA Jetson Platform. They require JetPack 4.2. He's also maintaining source build instructions. This PR adds links to the binaries and source build instructions to the README. The links are dynamic, so when new stable / weekly wheels are available, Dustin will update the same URL to point to the new files Pull Request resolved: https://github.com/pytorch/pytorch/pull/18990 Differential Revision: D14820158 Pulled By: soumith fbshipit-source-id: 761a56557decb72ad9c1b9f8a2745667f558eec3
This commit is contained in:
committed by
Facebook Github Bot
parent
bcd527190a
commit
8961ad8c5b
18
README.md
18
README.md
@ -132,6 +132,21 @@ There is no wrapper code that needs to be written. You can see [a tutorial here]
|
||||
Commands to install from binaries via Conda or pip wheels are on our website:
|
||||
[https://pytorch.org](https://pytorch.org)
|
||||
|
||||
|
||||
#### NVIDIA Jetson platforms
|
||||
|
||||
Python wheels for NVIDIA's Jetson Nano, Jetson TX2, and Jetson AGX Xavier are available via the following URLs:
|
||||
|
||||
- Stable binaries:
|
||||
- Python 2.7: https://nvidia.box.com/v/torch-stable-cp27-jetson-jp42
|
||||
- Python 3.6: https://nvidia.box.com/v/torch-stable-cp36-jetson-jp42
|
||||
- Rolling weekly binaries:
|
||||
- Python 2.7: https://nvidia.box.com/v/torch-weekly-cp27-jetson-jp42
|
||||
- Python 3.6: https://nvidia.box.com/v/torch-weekly-cp36-jetson-jp42
|
||||
|
||||
They requires JetPack 4.2 and above and are maintained by @dusty-nv
|
||||
|
||||
|
||||
### From Source
|
||||
|
||||
If you are installing from source, we highly recommend installing an [Anaconda](https://www.continuum.io/downloads) environment.
|
||||
@ -146,6 +161,9 @@ If you want to compile with CUDA support, install
|
||||
If you want to disable CUDA support, export environment variable `NO_CUDA=1`.
|
||||
Other potentially useful environment variables may be found in `setup.py`.
|
||||
|
||||
If you are building for NVIDIA's Jetson platforms (Jetson Nano, TX1, TX2, AGX Xavier), Instructions to [are available here](https://devtalk.nvidia.com/default/topic/1049071/jetson-nano/pytorch-for-jetson-nano/)
|
||||
|
||||
|
||||
#### Install Dependencies
|
||||
|
||||
Common
|
||||
|
||||
Reference in New Issue
Block a user