运行 NVIDIA RTX Pro 4500/5000 时出错
作者: SaberLi110创建于 2026年1月13日更新于 2026年1月23日
大家好, 当我使用 Nvidia RTX PRO 4500/5000 时, 我修改了 Dockerfile, 相关信息如下: 1. 使用 nvidia/cuda:12.8.0-cudnn-devel-ubuntu22.04 镜像 2. 使用 jax[cuda12]==0.6.0 和 0.6.1 以及 0.8 3. 升级到 dm-haiku==0.0.14 4. 安装 tensorflow==2.12.0 和最新版本 我可以成功编译它, AlphaFold 运行了,但只使用 CPU;无法利用 GPU (nvidia-smi 显示没有进程)。 但是,如果我进入 AlphaFold 容器镜像, JAX 可以检测到 GPU。 有什么建议吗?
内容来源: google-deepmind/alphafold