I want test GPU is correctly work on pytorch
so i try run yolov5
but it dosen’t work
‘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.’
i try torchvision 0.15.2, 0.15.1, 0.15.0 because that link(git hub) say torch 2.0.0 is use 0.15
what is proper version for torch 2.0.0+nv23.5
or i reinstall torch older version?