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

检查here您应该根据您的 PyTorch 安装的版本。