Pytorch and Torchvision incompatibility in Ubuntu

[Calebtrueblue]

Thanks in advance for any help. I’m using Ubuntu 22.04 for this.

The error I’m getting is 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.

Been trying to fix this for a couple of days with no luck.

Torch is 3.1.2+Cuda12.1
Torchvision is 0.16.2+Cuda12.1
Python is 3.10.12

Compatibility Matrix shows this should work. Nvidia Drivers:

NVRM version: NVIDIA UNIX x86_64 Kernel Module 535.154.05 Thu Dec 28 15:37:48 UTC 2023
GCC version: gcc version 12.3.0 (Ubuntu 12.3.0-1ubuntu1~22.04)

Is there other information needed to help solve this? I’m relatively new to all of this.