先创建一个虚拟环境,安装python3.7版本

conda create --prefix=/lnteam/LC/anaconda3/envs/LCDGL python=3.7

浪潮V100环境配置pytorch环境和相应的DGL环境以及其他相关的库_实验环境

为新建的虚拟环境命名

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

浪潮V100环境配置pytorch环境和相应的DGL环境以及其他相关的库_DGL_02

安装DGL只能匹配torch的版本为0.5.3

pip install dgl-cu101==0.5.3

如果还要安装transformers可以安装4.1.1

浪潮V100环境配置pytorch环境和相应的DGL环境以及其他相关的库_DGL_03

pip install transformers==4.1.1

安装ignite

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pytorch-ignite