You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
OS: Windows 10
Graphics card: RTX3070
CUDA: 11.1 (also tried 11.0, see below)
Python: 3.7
I'm running deeplabcutcore with tensorflow 2.0 due to the issue of series 3000 cards not working with Tensorflow 1.x. I'm using a conda virtual environment and have tf-nightly-gpu '2.5.0-dev20210111' installed.
I'm new to using deeplabcut with a local GPU rather than colab and I'm having trouble with getting the GPU to engage during training.
When I test GPU with tf.config.list_physical_devices
yes, please see this issue and resolution here: #944 (i.e, would be good to post there, or in deeplabcutcore, as that is the code you are using, vs. new issue here) & please note the you need to use the correct CUDA for 2.5, which is not CUDA 10; please carefully check your driver, CUDA, and TF versions: https://www.tensorflow.org/install/source#gpu
I would never recommend using the nightly build of TF until you know you have an env. that works; please try a stable release:
OS: Windows 10
Graphics card: RTX3070
CUDA: 11.1 (also tried 11.0, see below)
Python: 3.7
I'm running deeplabcutcore with tensorflow 2.0 due to the issue of series 3000 cards not working with Tensorflow 1.x. I'm using a conda virtual environment and have tf-nightly-gpu '2.5.0-dev20210111' installed.
I'm new to using deeplabcut with a local GPU rather than colab and I'm having trouble with getting the GPU to engage during training.
When I test GPU with tf.config.list_physical_devices
I also tried using CUDA 11.0 instead, and then GPU was engaged but it crashed with CUBLAS_STATUS_ALLOC_FAILED error...
Any suggestions on the best way to proceed? Thanks!
The text was updated successfully, but these errors were encountered: