DeepLabCut使用问题汇总
本来不想记录的。。。
本计算机配置如下:
显卡2060,安装cuda 10.0和cudnn 7.6+版
tensorflow==1.15.0
- 特别说明,tensorflow只能是1.x.x 系列,不然会出现各种问题
- 如果tensorflow安装了2.x版本,就会出现
- 没有 contrib 等函数。。
1.cudnn不能启动
报错代码:
-
在 deeplabcut.create_labeled_video()中显示以下错误,导致无法生成video的分析数据,影响后面生成视频
Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED
解决方法:
在运行前添加代码,即可解决os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true' ### +
总结,之前出现了很多问题,各种gpu不能使用的问题,这里懒得记录了。一般加上 os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true' ### +
就可以了!!