单卡N260部署AlphaFold

单卡N260部署AlphaFold，是否有相关镜像可以提供，或者相关操作流程与步骤

尊敬的开发者您好，请使用沐曦开发者社区pytorch镜像，在镜像里安装mcjax软件包developer.metax-tech.com/softnova/search?package_name=maca-jax-0.4.34-py312-3.7.2.0-x86_64.tar.xz

可以在docker 的pytorch镜像环境中，使用jax进行启动可以吗
python run_alphafold.py \
--json_path=test.json \
--model_dir=models \
--output_dir=output \
--db_dir="" \
--run_data_pipeline=false

尊敬的开发者您好，可以启动

jaxlib.xla_extension.XlaRuntimeError: RESOURCE_EXHAUSTED: Shared memory size limit exceeded: requested 98304, available: 65536
会出现共享内存不足的问题，请问有解决方案吗

尊敬的开发者您好，请给出容器启动命令

python run_alphafold.py \
--json_path=test.json \
--model_dir=/data/alphafold3/models \
--output_dir=./output \
--norun_data_pipeline \
--flash_attention_implementation=xla

尊敬的开发者您好，是docker run命令

现在是在conda虚拟环境中，没有用docker

尊敬的开发者您好，请参考developer.metax-tech.com/forum/t/fa-tie-qian-bi-kan-jing-xiang-shi-yong-wen-ti-ti-wen-mo-ban/267/ 给出环境信息以及详细日志