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

然后就可以运行程序了

参考链接

相关推荐
Ainnle4 分钟前
企业级RAG开源项目分享:Quivr、MaxKB、Dify、FastGPT、RagFlow
人工智能·开源
Multiple-ji7 分钟前
想学python进来看看把
开发语言·python
liuyuzhongcc37 分钟前
List 接口中的 sort 和 forEach 方法
java·数据结构·python·list
小天努力学java38 分钟前
AI赋能传统系统:Spring AI Alibaba如何用大模型重构机票预订系统?
人工智能·spring
鸟哥大大1 小时前
【Python】pypinyin-汉字拼音转换工具
python·自然语言处理
jiugie1 小时前
MongoDB学习
数据库·python·mongodb
Fuweizn1 小时前
在工业生产中,物料搬运环节至关重要,搬运机器人开启新篇章
人工智能·智能机器人·复合机器人
技术员阿伟1 小时前
《AI赋能星际探索:机器人如何开启宇宙新征程!》
人工智能
技术员阿伟1 小时前
《解锁AI密码,机器人精准感知环境不再是梦!》
人工智能
阿尔法波1 小时前
python与pycharm如何设置文件夹为源代码根目录
开发语言·python·pycharm