(原创,转载请注明出处)
1、升级显卡驱动为
https://developer.nvidia.com/cuda-downloads
https://developer.download.nvidia.cn/compute/cudnn/redist/cudnn/linux-x86_64/
NVIDIA-SMI 565.57.01 Driver Version: 565.57.01 CUDA Version: 12.7
2、安装部署一下环境
source /appsnew/source/cmake-3.14.3.sh
source /appsnew/source/intel2022.sh
source /appsnew/source/gcc-12.1.0.sh
source /appsnew/source/cuda-12.6.2.sh
# change to your conda environment
source /appsnew/source/Anaconda3-2024.06-1-local.sh
conda create -n AF3 python=3.11
conda activate AF3
2、下载alphafold3
mkdir ./hmmer_build ./hmmer
wget http://eddylab.org/software/hmmer/hmmer-3.4.tar.gz --directory-prefix ./hmmer_build
cd ./hmmer_build && tar zxf hmmer-3.4.tar.gz && rm hmmer-3.4.tar.gz
cd ./hmmer-3.4 & ./configure --prefix $(realpath ../../hmmer)
cd ./easel && make install
pip3 install -r dev-requirements.txt
# if you failed in build pybind11
# try to manually install it!
# 或者梯子(clash)安装
python run_alphafold.py --helpfull #测试
(感谢朱jintao首次编译和测试)
提交脚本注意v100/L40的使用xla,程序默认为triton,v100/L40不能使用(还有cudnn选项),
export XLA_CLIENT_MEM_FRACTION=0.95
export XLA_CLIENT_MEM_FRACTION=3.2
--flash_attention_implementation=xla
AF3RUN.sh 已经加入判断了nvidia-smi --format=csv --query-gpu=name|grep -qi 'v100\|l40':
在北极星的提交方法:
pkurun-l4011 AF3RUN.sh1tce.json
pkurun-h80011AF3RUN.sh1tce.json
pkurun-a80011AF3RUN.sh1tce.json
(v100目前不支持,会爆炸 gpu_2l gpu_4l 分区)
上诉命令的提交脚本
[chenf@login28testclass]# cat job.srp185320
#SBATCH -o AF3185320_%j.out
#SBATCH -e AF3185320_%j.err
pkurun AF3RUN.sh8ujo.json