I've been struggling for days getting a rhel8 server with a tesla to work properly with python. All the Cuda samples and graphics driver work, but none of Jax, tf, or pytorch are letting me coerce them into finding the gpu. Maybe this will help?
They have others for tensorflow etc. I would highly recommend this as the host system only has to have the NVIDIA driver installed with the rest of the dependencies contained in the docker.
12 comments
[ 3.0 ms ] story [ 46.6 ms ] threadHere is the pytorch one: https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
They have others for tensorflow etc. I would highly recommend this as the host system only has to have the NVIDIA driver installed with the rest of the dependencies contained in the docker.