当我导入torchvision时，我得到一个错误，指出pytorch和torchvision的cuda版本不同

Question

导入torchvision时出现以下错误。

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

如何将pytorch的cuda版本改为10.1？

‘'conda install pytorch torchvision cudatoolkit=10.1-c pytorch’从anaconda提示我得到同样的错误，即使我这样做了。

我使用的是windows10，在jupyter notebook的虚拟环境中使用的是Python3.7版。

Answer 1

卸载并重新安装可能会起作用

!conda uninstall pytorch torchvision

通过以下方式：

!conda install pytorch torchvision cudatoolkit=10.1 -c pytorch