首先安装步骤可以参考
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