meta-pytorch / meta-pytorch/torchcodec

Unsupported device: cuda (device type: cuda, variant: beta)

Open
#1,043 2 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
1.2k
Forks
125
Avg merge
22h 47m
Merged PRs (30d)
54

Description

I just built the master branch on NVIDIA Jetson Thor (Blackwell, CUDA 13.0, CUDA Arch 11.0), but I get the following issue with the version set specified below:

torch=2.9.0a0+50eac811a6.nv25.9
torchvision=0.24.0a0+98f8b375
torchcodec=0.9.0a0

Test code:

rom torchcodec.decoders import set_cuda_backend, VideoDecoder

video_file = "video.mp4"

with set_cuda_backend("beta"):  # Use the BETA backend, it's faster!
    decoder = VideoDecoder(video_file, device="cuda")

Error:

Traceback (most recent call last):
  File "/root/test_codec.py", line 6, in <module>
    decoder = VideoDecoder(video_file, device="cuda")
              ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/usr/local/lib/python3.12/dist-packages/torchcodec/decoders/_video_decoder.py", line 154, in __init__
    core.add_video_stream(
  File "/usr/local/lib/python3.12/dist-packages/torch/_ops.py", line 840, in __call__
    return self._op(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: Unsupported device: cuda (device type: cuda, variant: beta)

As it is already specified with RTX 5090 here https://github.com/meta-pytorch/torchcodec/issues/1006#issuecomment-3485482616, I guess it is related to Blackwell architecture. Is there any workaround to this?

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

Reproduce the failure with the shown VideoDecoder and set_cuda_backend("beta") example, then inspect torchcodec/decoders/_video_decoder.py around VideoDecoder and the core.add_video_stream call. Compare the behavior with the RTX 5090 discussion in issue #1006. Done means identifying whether Jetson Thor/Blackwell is unsupported and documenting a verified workaround or limitation.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
audio-video-rtc
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.