Torch and torchvision version cu mismatch

vision
1

As near as I can tell, my system has torch-torchvision version mismatch. Hope this is the right place to ask.

I received this error running a few ComfyUI nodes (FaceDetailer, ExpressionEditor)..

Could not run ‘torchvision::nms’ with arguments from the ‘CUDA’ backend.

windows 10, nvidia 3090, python 3.10.6

torch version: 2.7.0+cu128

torchvision version: 0.22.0+cpu

although, nvidia-smi says cuda version 12.6

my questions:

should I uninstall torch and torchvision first before reinstalling them?

is there some way to revert if this runs into trouble?

thanks

2

Try to uninstall torchvision and try to reinstall it via pip install torchvision==0.22.0 --index-url ``https://download.pytorch.org/whl/cu128.

This should install the right vision wheel with CUDA 12.8 support matching PyTorch.