diff --git a/.ci/docker/requirements.txt b/.ci/docker/requirements.txt index 58cbcb304d..daf515b6a7 100644 --- a/.ci/docker/requirements.txt +++ b/.ci/docker/requirements.txt @@ -31,6 +31,7 @@ librosa torch==2.14 cuda-bindings>=13.1.0 # Required for CUDA graph annotations tutorial torchvision +torchcodec torchdata networkx PyHamcrest diff --git a/beginner_source/basics/data_tutorial.py b/beginner_source/basics/data_tutorial.py index c49f25a587..937768a311 100644 --- a/beginner_source/basics/data_tutorial.py +++ b/beginner_source/basics/data_tutorial.py @@ -120,7 +120,7 @@ import os import pandas as pd -from torchvision.io import decode_image +from torchcodec.decoders import decode_image class CustomImageDataset(Dataset): def __init__(self, annotations_file, img_dir, transform=None, target_transform=None):