在启智平台安装dgl库

2023-12-31 16:36:17

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
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。