1.因为九天平台默认的cuda版本是10.1,而dgl要求的最低适配的cuda版本是10.2,所有需要自己创建虚拟环境,安装对应版本的cuda
创建虚拟环境:
conda create -n 虚拟环境名字 python=3.7
安装cuda:
conda install cudatoolkit=10.2
conda install cudnn=7.6.5
退出虚拟环境,执行下面这个命令:
conda env config vars set LD_LIBRARY_PATH=/root/.local/conda/envs/虚拟环境名字/lib -n 虚拟环境名字
再进入虚拟环境中,
安装对应版本torch:
# cuda10.2版本,torch1.8版本
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch
查看cuda的版本:
import torch
print(torch.version.cuda)
2.安装dgl库
在下面这个链接中查找对应版本的压缩包,下载到本地,传到操作台中
https://conda.anaconda.org/dglteam/linux-64
解压:
tar -jxvf 压缩包名字
查看系统中torch的安装位置:
python -c 'import site; print(site.getsitepackages()[0])'
输出:/root/.local/conda/envs/new/lib/python3.7/site-packages
cp -r lib/python3.7/site-packages/dgl* /root/.local/conda/envs/new/lib/python3.7/site-packages
还需要安装scipy、networkx。
pip install scipy
pip install networkx