Pytorch和Torchvision编译不同的CUDA版本

Question

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.

我尝试在我的 Google Colab 上运行 YOLACT 并发现此错误。有人可以帮助解决这个问题吗？

Answer 1

您需要升级torchvision到使用 CUDA 10.2 编译的版本：

pip install --upgrade torchvision>=0.6.0

或者，如果您使用 Conda：

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

在此处检查您应根据您的 PyTorch 安装的版本。