5 ms·

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,

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?

uneekname 5y ago
Oof, i feel your pain. Yes, it sounds like this might help.
thebruce87m 5y ago
Have you tried any of NVIDIAs docker containers? Here is the pytorch one: https://ngc.nvidia.com/catalog/containers/nvidia:pytorch 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.
disgruntledphd2 5y ago
Ld load path maybe? I remember this happened to me once a few years back?