在原有pytorch环境下安装DGL库和其对应的CUDA【自用】

前段时间看到一篇AAAI2024的论文Patch-wise Graph Contrastive Learning for Image Translation,它采用GNN的思想来进行image-to-image translation的任务,非常的新颖,但我进行复现的时候,发现直接下载它里面需要的DGL库是无法运行的,总是说缺少对应的CUDA。
因此我们需要重新创建一个环境,安装DGL库和其对应的CUDA。

1. 创建pytorch环境
打开Anaconda Navigator,创建环境,命名为DGL,python版本选为3.7.

DGL有CPU版本和CUDA版本
CPU版本

python 复制代码
https://pypi.org/simple/dgl/

CUDA版本

python 复制代码
https://anaconda.org/dglteam/repo

我们选择是CUDA版本,打开网站后选择自己对应的CUDA版本

我们选择 win-64/dgl-cuda11.1-0.9.1-py37_0.tar.bz2,进行下载

打开Pycharm,加载解释器为DGL

打开终端,输入

python 复制代码
conda install xxx(路径)/win-64/dgl-cuda11.1-0.9.1-py37_0.tar.bz2

然后就可以运行程序了

参考链接

相关推荐
陈广亮5 分钟前
MCP 2026路线图解读.md
人工智能
找藉口是失败者的习惯5 分钟前
【Android】Android 车机 + AI Agent 有没有搞头?
android·人工智能
东离与糖宝6 分钟前
Langflow-ai OpenRAG实战:Java+Spring Boot搭建企业级私有知识库(从0到1)
java·人工智能
Rsun045517 分钟前
Spring AI 面试题
java·人工智能·spring
星浩AI7 分钟前
OpenClaw 总是忘记你说过的话?来试试这个插件
人工智能·后端·agent
前端付豪10 分钟前
AI知识库 + RAG数学解析增强
前端·python·llm
tinygone10 分钟前
OpenClaw+FunASR识别飞书发来的音频文件
人工智能·飞书
风停又起10 分钟前
智能体Agent从 0 到 OpenClaw:AI Agent 的完整演进之路
人工智能