protein_generator安装

conda create -n pg python==3.8.16

conda activate pg


# dgl需求的torch版本不一致，需要官网查看，这里使用2.1和cuda-118，不支持5090
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118
pip install dgl -f https://data.dgl.ai/wheels/torch-2.1/cu118/repo.html
pip install e3nn==0.5.1
pip install biopython==1.81
pip install icecream==2.1.3
pip install seaborn==0.12.2
pip install pynvml==11.0.0
pip install decorator==5.1.0