GPU-Support available?

[RaoulusHPC]

Hi,

I'm just curious if your images are already providing GPU support? Whenever I start TF 2.4.1 within the container, I'll get the following promt:

2021-05-05 20:27:55.185615: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/nvidia/lib:/usr/local/nvidia/lib64 2021-05-05 20:27:55.185670: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.

Thanks for your help!