In [1]: import torchvision
---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
Cell In[1], line 1
----> 1 import torchvision

File /usr/lib/python3.11/site-packages/torchvision/__init__.py:6
      3 from modulefinder import Module
      5 import torch
----> 6 from torchvision import datasets, io, models, ops, transforms, utils
      8 from .extension import _HAS_OPS
     10 try:

File /usr/lib/python3.11/site-packages/torchvision/datasets/__init__.py:1
----> 1 from ._optical_flow import FlyingChairs, FlyingThings3D, HD1K, KittiFlow, Sintel
      2 from ._stereo_matching import (
      3     CarlaStereo,
      4     CREStereo,
   (...)
     12     SintelStereo,
     13 )
     14 from .caltech import Caltech101, Caltech256

File /usr/lib/python3.11/site-packages/torchvision/datasets/_optical_flow.py:12
      9 import torch
     10 from PIL import Image
---> 12 from ..io.image import _read_png_16
     13 from .utils import _read_pfm, verify_str_arg
     14 from .vision import VisionDataset

File /usr/lib/python3.11/site-packages/torchvision/io/__init__.py:8
      5 from ..utils import _log_api_usage_once
      7 try:
----> 8     from ._load_gpu_decoder import _HAS_GPU_VIDEO_DECODER
      9 except ModuleNotFoundError:
     10     _HAS_GPU_VIDEO_DECODER = False

File /usr/lib/python3.11/site-packages/torchvision/io/_load_gpu_decoder.py:1
----> 1 from ..extension import _load_library
      4 try:
      5     _load_library("Decoder")

File /usr/lib/python3.11/site-packages/torchvision/extension.py:107
    102             warn("LoadLibraryExW is missing in kernel32.dll")
    104     torch.ops.load_library(lib_path)
--> 107 _check_cuda_version()

File /usr/lib/python3.11/site-packages/torchvision/extension.py:80, in _check_cuda_version()
     78     t_minor = int(t_version[1])
     79     if t_major != tv_major or t_minor != tv_minor:
---> 80         raise RuntimeError(
     81             "Detected that PyTorch and torchvision were compiled with different CUDA versions. "
     82             f"PyTorch has CUDA Version={t_major}.{t_minor} and torchvision has "
     83             f"CUDA Version={tv_major}.{tv_minor}. "
     84             "Please reinstall the torchvision that matches your PyTorch install."
     85         )
     86 return _version

RuntimeError: Detected that PyTorch and torchvision were compiled with different CUDA versions. PyTorch has CUDA Version=12.2 and torchvision has CUDA Version=12.1. Please reinstall the torchvision that matches your PyTorch install.