Error module torchvision in CUDA 11.4

torch.compile
1

can i ask im install torch version 1.12.0a0+2c916ef.nv22.3 and install torchvision 0.13 in CUDA 11.4 but not compatible, can help me for solution?

Python 3.8.10 (default, Sep 11 2024, 16:02:53)
[GCC 9.4.0] on linux
Type “help”, “copyright”, “credits” or “license” for more information.

import torch
import torchvision
/home/roisc-orin1/.local/lib/python3.8/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension:
warn(f"Failed to load image Python extension: {e}")

print(f"PyTorch Version: {torch.version}“)
PyTorch Version: 1.12.0a0+2c916ef.nv22.3
print(f"Torchvision Version: {torchvision.version}”)
Torchvision Version: 0.13.0

2

Could you describe what kind of error is pointing to the incompatibility?

3

like this

home/roisc-orin1/.local/lib/python3.8/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension:
warn(f"Failed to load image Python extension: {e}")

Traceback (most recent call last):
File “yolo.py”, line 27, in
results = model.predict(source=resized_frame)
File “/home/roisc-orin1/.local/lib/python3.8/site-packages/torch/autograd/grad_mode.py”, line 27, in decorate_context
return func(*args, **kwargs)
File “/home/roisc-orin1/.local/lib/python3.8/site-packages/ultralytics/yolo/engine/model.py”, line 253, in predict
return self.predictor.predict_cli(source=source) if is_cli else self.predictor(source=source, stream=stream)
File “/home/roisc-orin1/.local/lib/python3.8/site-packages/ultralytics/yolo/engine/predictor.py”, line 184, in call
return list(self.stream_inference(source, model)) # merge list of Result into one
File “/home/roisc-orin1/.local/lib/python3.8/site-packages/torch/autograd/grad_mode.py”, line 43, in generator_context
response = gen.send(None)
File “/home/roisc-orin1/.local/lib/python3.8/site-packages/ultralytics/yolo/engine/predictor.py”, line 244, in stream_inference
self.results = self.postprocess(preds, im, im0s)
File “/home/roisc-orin1/.local/lib/python3.8/site-packages/ultralytics/yolo/v8/pose/predict.py”, line 16, in postprocess
preds = ops.non_max_suppression(preds,
File “/home/roisc-orin1/.local/lib/python3.8/site-packages/ultralytics/yolo/utils/ops.py”, line 246, in non_max_suppression
i = torchvision.ops.nms(boxes, scores, iou_thres) # NMS
File “/home/roisc-orin1/.local/lib/python3.8/site-packages/torchvision/ops/boxes.py”, line 40, in nms
_assert_has_ops()
File “/home/roisc-orin1/.local/lib/python3.8/site-packages/torchvision/extension.py”, line 33, in _assert_has_ops
raise RuntimeError(
RuntimeError: Couldn’t load custom C++ ops. This can happen if your PyTorch and torchvision versions are incompatible, or if you had errors while compiling torchvision from source. For further information on the compatible versions, check GitHub - pytorch/vision: Datasets, Transforms and Models specific to Computer Vision for the compatibility matrix. Please check your PyTorch version with torch.version and your torchvision version with torchvision.version and verify if they are compatible, and if not please reinstall torchvision so that it matches your PyTorch install.