重要的话说三遍:
       1.用conda下载!用conda下载!用conda下载!

       2.dgl和torch用官网下!dgl和torch用官网下!dgl和torch用官网下!

      3.不要使用清华镜像源下载,后端下载torch-gpu版本,其实还是cpu版本。


pip安装遇到的报错

1FileNotFoundError: Cannot find DGL C++ graphbolt library at C:\Users\13043\AppData\Local\Programs\Python\Python311\Lib\site-packages\dgl\graphbolt\graphbolt_pytorch_2.4.1.dll

2

我们论文复现的是要求的环境是

Prerequisites:**
- Python 3.8+
- PyTorch 1.8+
- DGL 0.8+ (Follow installation guide [here](https://www.dgl.ai/pages/start.html))
(提供了DGL官网)
- CUDA 11.0+ (if using GPU)

我选择了自己电脑安装

Cuda12.1

Anoconda配置3.9的虚拟环境

Dgl2.0.0.cu121

Pytorch2.2.1


大家自行安装anoconda,cuda12.1

1.pycharm终端配置一个conda的虚拟环境

#终端创建conda虚拟环境
conda create --name torchyanshi python=3.9 --no-default-packages
#conda激活
conda activate torchyanshi

设置-添加解释器-添加本地解释器-conda环境 将其激活

如果后续有问题记得重启pycharm

电脑已经安装了cuda12.1

2先安装dgl再安装torch,通过dgl官网https://www.dgl.ai/pages/start.html

兼容torch2.2,cuda12.1的dgl
conda install -c dglteam/label/cu121 dgl

3.官网安装pytorch2.2 Previous PyTorch Versions | PyTorch

安装torch选择合适的节点快一些

conda install pytorch==2.2.1 torchvision==0.17.1 torchaudio==2.2.1 pytorch-cuda=12.1 -c pytorch -c nvidia

有问题的话记得重启pycharm并且conda更新索引会很慢,大家等一会。

导入库的时候就发现什么问题都没有了,真的调了两天的环境。哇,dgl对小白真的不友好啊。

import torch
import dgl
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.cuda.current_device())
print(torch.cuda.get_device_name(0))
import torch
print(torch.__version__)
print(torch.cuda.is_available())
print(torch.version.cuda)

True
1
0
NVIDIA GeForce GTX 1650
2.2.1
True
12.1
#dgl可以安装,不过有个模块缺失了。
from DKG.model import DynamicGraphModel, DKG_DEFAULT_CONFIG, EmbeddingUpdater, Combiner, EdgeModel, InterEventTimeModel, MultiAspectEmbedding

ModuleNotFoundError: No module named 'torch_scatter'

#根据电脑的torch版本和cuda版本安装兼容的
pip install torch-scatter -f https://data.pyg.org/whl/torch-2.2.0+cu121.html

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐