pytorch系列(三)——cuda下DGL库安装

1、不限制DGL安装版本时

进入DGL官网
根据cuda版本选择对应的命令
pytorch系列(三)——cuda下DGL库安装_第1张图片复制Run this 后面这句话:conda install -c dglteam dgl-cuda10.2 到终端,按下回车等待出现是否同意继续,输入y。

在这里插入图片描述
写文章时是默认自动安装0.7.1版本。

若有指定DGL版本的需求,请看2内容。

2、指定DGL版本

若已经安装了不需要的DGL版本,比如0.7.1想替换为0.4.3版本,则要删除现有版本:

默认删除当前DGL:conda uninstall -c dglteam dgl-cuda10.2
(cuda10.2可根据自己的环境调整)
指定删除版本:conda uninstall -c dglteam dgl-cuda10.2==0.5.0
(cuda10.2==0.5.0可根据自己的环境调整,使用 conda list 可以查看使用的当前版本)

若尚未安装DGL:

直接指定版本即可, 例如:
conda install -c dglteam dgl-cuda10.2==0.4.3post2
查看dgl版本名称网站

你可能感兴趣的:(pytorch,linux,python)