pytorch / pytorch/vision

For the nvidia jetson package , the pytorch and torchvision version compatibility does not work if you follow the compatibility table

Open
#7,076 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
17.9k
Forks
7.3k
Avg merge
1d 15h
Merged PRs (30d)
13

Description

📚 The doc issue

When a user has to install something like detectron2 , the user needs torch, torchvision and cuda tool kit. The version for each must be compatible. The compatibility table works very well for Desktop or Linux based machines with GPU. The compatibility table does not work for arm architectures and this information is not provided anywhere

Suggest a potential alternative/fix

A suggestion is to probably have the table modified for arm based architectures.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

No file or test is named in the issue. First locate the compatibility table and verify the PyTorch, torchvision, and CUDA combinations for NVIDIA Jetson ARM devices; done means the documentation clearly includes supported ARM-specific versions or points users to authoritative compatibility information.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
documentation, embedded-iot, machine-learning
Issue type
Documentation
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.