2021年再次使用deeplabcut遇到的坑记录

首先安装步骤可以参考

deeplabcut的官方说明

1.AttributeError: module 'tensorflow' has no attribute 'unravel_index'

在deeplabcut==2.1.8版本下的报错，这个只有tensorflow-gpu==1.15或者2.5.0有

一开始装了cuda11.1和tensorflow-gpu==2.5.0，结果各种库不齐全，后来看了很多大佬说2.5.0还是要用cuda11.2，感觉还是用低版本试试好了

 

2.报错cudnn无法使用

UnknownError: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [Op:Conv2D]

晚上有很多人说这样可以解决，但是我尝试后并没有成功【大家也可以先看看cuda版本是否对应成功】，后来我发现是因为我用的deeplabcut==2.1.8的版本的问题

#避免报错cudnn无法使用
os.environ['TF_FORCE_GPU_ALLOW_GROWTH'] = 'true'

最后成功的环境配置：

cuda 9.0  

tensorflow-gpu==1.8

python=3.6

deeplabcut==2.1.9【这个版本要注意，我感觉我的坑都在此处】

wxpython==4.0.4