Compatibility with RTX 3080 about deeplabcut-core HOT 4 OPEN

OS: Win 10
Installation sequence:
Build cuda following:
https://www.reddit.com/r/tensorflow/comments/jsalkw/rtx_3090_and_tensorflow_for_windows_10_step_by/
Cuda: 11.1
cuDNN: v8.0.5.39
Anaconda env used: DLC-GPU (latest, v2.1.10.2)
Tensorflow Version: pip install tensorflow-gpu==2.4.1
Deeplabcutcore: pip installed first, later from github directly (see below)
tf-slim: pip install tf-slim==1.1.0

Hi all,
Another RTX 3070 user checking in.

I found out that the pip installed deeplabcutcore seems to be an older version than the github one.
When I ran the testscript.py with the pip installed deeplabcutcore, it gives a lot of errors related to tensorflow library while importing deeplabcutcore (for example, module 'tensorflow.python.framework.ops' has no attribute 'RegisterShape').
But if I download the github repo, it works.

Plus, I had the same problem with VRAM exploding, too.
I guess it might be because I am using RTX 3070 as my displaying GPU, too.
I found a solution in here:
tensorflow/tensorflow#46209
I added the following two lines after "import tensorflow as tf" in Lib\site-packages\deeplabcutcore\pose_estimation_tensorflow\train.py:
physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True).
It then worked well.

Hope it helps!
Best,
Chen

from deeplabcut-core.

Hello. I'm a RTX3000 user too!

This may not be the answer you are looking for, but I was able to run DeepLabCut 2.2b8 on my RTX3080 using the package tensorlow-directml.
This is an UNOFFICIAL WAY with a package not originally used by DeepLabCut, and it works slower than using native CUDA, but I was able to confirm that it works.

I wrote a report about this on imaga.sc: https://forum.image.sc/t/fyi-deeplabcut-worked-on-radeon-gpu-rtx3080-using-tensorflow-directml/47700

from deeplabcut-core.

Thanks for this post -- I have also been having these exact issues when trying to install with RTX 3070.
I made an initial post which didn't go into the same level of detail as you have: DeepLabCut/DeepLabCut#1078
I would love to hear if you find out how to get it running!

from deeplabcut-core.

Hi Yao, I am new to DLC and I also use RTX2060 but it does not work. Can you share your config? Best

from deeplabcut-core.