Open forcefield opened 8 months ago
If I build the docker using the provided docker/Dockerfile, with the following two lines
docker/Dockerfile
ARG CUDA=11.8.0 FROM nvidia/cuda:${CUDA}-cudnn8-runtime-ubuntu22.04
When I run docker/run_docker.py, I get the error
docker/run_docker.py
I0125 13:11:56.513862 140328062711616 run_docker.py:258] jaxlib.xla_extension.XlaRuntimeError: INTERNAL: Could not find the corresponding function
But if I build using the cuda with the devel label by changing the second line of the above to
devel
FROM nvidia/cuda:${CUDA}-cudnn8-devel-ubuntu22.04
The docker runs fine. So something is missing in the linked library when building using cuda with the runtime label.
runtime
If I build the docker using the provided
docker/Dockerfile, with the following two linesWhen I run
docker/run_docker.py, I get the errorI0125 13:11:56.513862 140328062711616 run_docker.py:258] jaxlib.xla_extension.XlaRuntimeError: INTERNAL: Could not find the corresponding functionBut if I build using the cuda with the
devellabel by changing the second line of the above toThe docker runs fine. So something is missing in the linked library when building using cuda with the
runtimelabel.