Cornernet-lite 环境配置

最新推荐文章于 2024-08-10 07:57:43 发布

Sesen_s

最新推荐文章于 2024-08-10 07:57:43 发布

阅读量3.3k

点赞数 2

分类专栏：学习笔记 Linux工程问题深度学习

本文链接：https://blog.csdn.net/weixin_40546602/article/details/90200247

版权

在尝试运行CornerNet_lite的Python脚本时，遇到了多个错误，包括：导入错误，CUDA版本过旧，NVIDIA驱动问题以及torch.cuda.is_available()返回False。错误提示表明需要更新NVIDIA驱动至合适版本，解决CUDA版本不匹配问题，以及可能需要调整PyTorch和CUDA的版本。已尝试降低PyTorch版本但未成功，最终通过卸载并重新安装特定版本的PyTorch和CUDA100来解决问题。

摘要由CSDN通过智能技术生成

错误1 :
运行Python demo.py error:
Traceback (most recent call last):
File "demo.py", line 6, in <module>
from core.detectors import CornerNet_Squeeze
File "/data/shiwei/CornerNet_Lite/core/detectors.py", line 1, in <module>
from .base import Base, load_cfg, load_nnet
File "/data/shiwei/CornerNet_Lite/core/base.py", line 3, in <module>
from .nnet.py_factory import NetworkFactory
File "/data/shiwei/CornerNet_Lite/core/nnet/py_factory.py", line 7, in <module>
from ..models.py_utils.data_parallel import DataParallel
File "/data/shiwei/CornerNet_Lite/core/models/py_utils/__init__.py", line 1, in <module>
from ._cpools import TopPool, BottomPool, LeftPool, RightPool
File "/data/shiwei/CornerNet_Lite/core/models/py_utils/_cpools/__init__.py", line 6, in <module>
import top_pool, bottom_pool,