Traceback (most recent call last): File "D:/SRP/源码/111/faster-rcnn-pytorch-master/predict.py", line 14, in <module> frcnn = FRCNN() File "D:\SRP\源码\111\faster-rcnn-pytorch-master\frcnn.py", line 80, in __init__ self.std = self.std.cuda() File "D:\Anaconda\Anaconda3\envs\shiyan\lib\site-packages\torch\cuda\__init__.py", line 239, in _lazy_init raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA enabled
时间: 2023-06-12 13:06:00 浏览: 447
rich-traceback:Python 日志记录模块的信息回溯
这个错误提示说明你的 PyTorch 没有启用 CUDA,也就是说你的 PyTorch 不能在 GPU 上运行。你需要检查你的 PyTorch 安装是否正确以及是否启用了 CUDA。如果你的 PyTorch 安装正确并已启用 CUDA,但仍然出现这个错误,你可以尝试重新安装 PyTorch,并确保你的 GPU 驱动程序已正确安装。
阅读全文