先创建一个虚拟环境,安装python3.7版本
conda create --prefix=/lnteam/LC/anaconda3/envs/LCDGL python=3.7
为新建的虚拟环境命名
conda config --append envs_dirs /lnteam/LC/anaconda3/envs/
V100的cuda版本为10.1所以下载pytorch版本为1.7
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 -f https://download.pytorch.org/whl/torch_stable.html
安装DGL只能匹配torch的版本为0.5.3
pip install dgl-cu101==0.5.3
如果还要安装transformers可以安装4.1.1
pip install transformers==4.1.1
安装ignite
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pytorch-ignite