Skip to content

Commit 2a3ba5d

Browse files
authored
Merge branch 'main' into 313
2 parents 03371d9 + e6391ff commit 2a3ba5d

File tree

11 files changed

+9
-994
lines changed

11 files changed

+9
-994
lines changed

test/test_video_gpu_decoder.py

Lines changed: 0 additions & 97 deletions
This file was deleted.

torchvision/__init__.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -57,16 +57,12 @@ def set_video_backend(backend):
5757
backend, please compile torchvision from source.
5858
"""
5959
global _video_backend
60-
if backend not in ["pyav", "video_reader", "cuda"]:
61-
raise ValueError("Invalid video backend '%s'. Options are 'pyav', 'video_reader' and 'cuda'" % backend)
60+
if backend not in ["pyav", "video_reader"]:
61+
raise ValueError("Invalid video backend '%s'. Options are 'pyav' and 'video_reader'" % backend)
6262
if backend == "video_reader" and not io._HAS_CPU_VIDEO_DECODER:
6363
# TODO: better messages
6464
message = "video_reader video backend is not available. Please compile torchvision from source and try again"
6565
raise RuntimeError(message)
66-
elif backend == "cuda" and not io._HAS_GPU_VIDEO_DECODER:
67-
# TODO: better messages
68-
message = "cuda video backend is not available."
69-
raise RuntimeError(message)
7066
else:
7167
_video_backend = backend
7268

torchvision/csrc/io/decoder/gpu/README.rst

Lines changed: 0 additions & 21 deletions
This file was deleted.

0 commit comments

Comments
 (0)