实验室服务器cuda更新为11.4
找的文献所需环境其中pytorch==1.11、dgl==0.9.0
最终配置的环境为python3.8、pytorch==1.12、dgl==1.0.0,也能跑代码
首先,尝试直接pip dgl==0.9.0以及从官网(Deep Graph Library (dgl.ai))上直接pip都会更新虚拟环境下的pytorch版本
求助善良可爱的同学后,有效下载方法:
先打开dgl官网
官网上没有cuda11.4对应指令,实际尝试下来,cuda11.4和cuda11.3相近,同时都可兼容pytorch1.12,
打开安装指令对应的网址data.dgl.ai/wheels/cu113/repo.html
可以看到cuda11.3对应最低版本为dgl1.0.0,我们所需0.9.0,1.0.0也能用
选择cu113为cuda11.3,cp38为python3.8,Linux64位版本,下载到本地,之后上传到服务器项目文件同目录下
然后在项目中进行pip下载即可
pip dgl-1.0.0+cu113-cp38-cp38-manylinux1_x86_64.whl