AttributeError: module 'torch._C' has no attribute 'nms'

引用: 运行main.py时报错如下：AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'。
您遇到的这个错误是由于使用的torch版本不兼容GPU导致的。根据您提供的引用的信息，您的torch版本适用于CPU而不适用于GPU。要解决这个问题，您可以考虑以下几个步骤：

1. 首先，确保您已经正确安装了与GPU兼容的torch版本。您可以通过以下命令检查torch是否支持GPU：

   import torch
   print(torch.cuda.is_available())

如果输出结果为False，说明您当前安装的torch版本不支持GPU。请卸载当前的torch版本，并重新安装与GPU兼容的版本。

2. 您可以通过以下命令卸载torch：

   pip uninstall torch

3. 然后，您可以根据您的CUDA版本选择合适的torch版本进行安装。您可以在pytorch官方网站上找到与您的CUDA版本兼容的torch版本。

4. 安装与您的CUDA版本兼容的torch版本。您可以使用以下命令进行安装（以torch 1.4.0为例）：

   pip install torch==1.4.0

5. 安装完成后，再次运行您的代码，应该就能够解决AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'错误了。