在启智平台安装dgl库

发布时间:2023年12月31日

1.输入nvcc- V查看CUDA的版本号

2.在这个网址上找到python版本、cuda版本、dgl版本都对应的包,复制对应的链接
https://conda.anaconda.org/dglteam/linux-64

3.通过wget进行安装,或者直接下载下来拖到服务器上

wget https://conda.anaconda.org/dglteam/linux-64/dgl-cuda10.2-0.7.2-py37_0.tar.bz2

4.解压

tar -jxvf dgl-cuda10.2-0.7.2-py37_0.tar.bz2 

5.通过这个命令查看系统中torch的安装位置

python -c 'import site; print(site.getsitepackages()[0])'

6.前面是dgl解压的位置,后面是torch的安装位置

cp -r lib/python3.7/site-packages/dgl* /opt/conda/lib/python3.7/site-packages/

7.测试dgl是否安装成功

import dgl
文章来源:https://blog.csdn.net/weixin_44021274/article/details/135316813
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。