Pytorch实现CPN
https://github.com/GengDavid/pytorch-cpn
配置问题记录
1.基础环境配置
-
使用anaconda来配置,pytorch 0.4,cudn 8.0, cudnn 7.1.3
-
opencv 3.4, sckit-image 0.13.4
2.问题记录
-
使用anaconda中opencv时,出现libstdc++.so.6.0.22找不到,导致opencv无法使用
问题:网上搜索,问题在于libstdc标准库版本太旧。
方法:下载对应6.0.22版本,安装在anaconda对应环境lib中,更新软链接 -
cocotools编译后仍然无法使用
问题:按照官方方法编译完成后,在python中能正常导入,但运行代码仍然报错
方法:问题在于,在anaconda子环境中编译和通用环境中编译是不能通用的,也就是想在anaconda特定环境中调用,必须在其子环境中编译,并在其python版本下验证是否能导入成功。 -
cuda无法使用
问题:RuntimeError: CUDNN_STATUS_INTERNAL_ERROR
方法:在运行代码前,强行将数据塞入gpu中
代码:在运行代码前加入CUDA_VISIBLE_DEVICES=0 CUDA_CACHE_PATH='~/.cudacache'