'Pytorch and Torchvision are compiled different CUDA versions

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

I am trying to run YOLACT on my Google Colab and found this error. Can someone help solve this issue?

pytorch torchvision

Solution 1:^[1]

You need to upgrade your torchvision to one compiled with CUDA 10.2:

pip install --upgrade torchvision>=0.6.0

or, if you're using Conda:

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

Check here the version you should install based on your PyTorch.

Sources

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Source: Stack Overflow

Solution	Source
Solution 1	Berriel

'Pytorch and Torchvision are compiled different CUDA versions

Solution 1:[1]

Sources

Related Questions

Solution 1:^[1]