在原有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

然后就可以运行程序了

参考链接

相关推荐
xingyuzhisuan几秒前
稳定性考验:连续跑7天,哪家云主机不重启、不掉线?
服务器·人工智能·gpu算力
sanshanjianke11 分钟前
AI辅助网文创作理论研究笔记(十):软件框架设计——模块化B/S架构
人工智能·ai写作
CHANG_THE_WORLD12 分钟前
python 批量终止进程exe
开发语言·python
liann11918 分钟前
3.2_红队攻击框架--MITRE ATT&CK‌
python·网络协议·安全·网络安全·系统安全·信息与通信
云天AI实战派28 分钟前
AI 智能体问题排查指南:ChatGPT、API 调用到 Agent 上线失灵的全流程修复手册
大数据·人工智能·python·chatgpt·aigc
Tutankaaa1 小时前
知识竞赛题库设计全攻略
人工智能·算法
我的xiaodoujiao1 小时前
API 接口自动化测试详细图文教程学习系列15--项目实战演练2
python·学习·测试工具·pytest
TImCheng06091 小时前
职场人AI学习周期评估:不同学习路径的时间成本
人工智能·学习
m0_466525291 小时前
酷特AGI:从“自家试验田”到“全球输出”
大数据·人工智能·agi
星爷AG I1 小时前
20-1 记忆概览(AGI基础理论)
人工智能·agi