图神经网络之DGL安装

使用简单的pip直接安装dgl是可以成功的,但训练使用cuda的话则会报错,注意,如果使用torch框架,则必须在dgl安装之前安装torch…

DGL安装:

conda install -c dglteam dgl-cuda10.1

此处cuda10.1必须指定,也可以其他版本

你可能感兴趣的:(笔记,DGL)