error running with a nvidia rtx pro 4500/5000
Author: SaberLi110Created Jan 13, 2026Updated Jan 23, 2026
Hi, everyone
When I was using the Nvidia RTX PRO 4500/5000, I modified the Dockerfile, and the relevant information is as follows:
- using the nvidia/cuda:12.8.0-cudnn-devel-ubuntu22.04 image
- using jax[cuda12]==0.6.0 and 0.6.1 and 0.8
- upgrading to dm-haiku==0.0.14
- installing tensorflow==2.12.0 and latest
I can successfully compile it, and AlphaFold runs, but only using the CPU; it cannot utilize the GPU (nvidia-smi shows no processes). However, if I enter the AlphaFold container image, JAX can detect the GPU.
any suggestions?
Source: google-deepmind/alphafold