conda(not pip)本地部署dgl,torch,cuda详细教程
·
重要的话说三遍:
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
更多推荐



所有评论(0)