当前位置: 首页 > news >正文

dgl 的cuda 版本 环境配置(dgl cuda 版本库无法使用问题解决)

1. 如果你同时有dgl  dglcu-XX.XX 那么,应该只会运行dgl (DGL的CPU版本),因此,你需要把dgl(CPU)版本给卸载了

但是我只卸载CPU版本还不够,我GPU 版本的dglcu依旧不好使,因此吧GPU版本的也得卸载了重新安装

最新版的dgl我的cuda版本已经不配了,因此,找老的版本:Linux 64 :: Anaconda.org

我下载的是这个:Dgl Cuda11.3 :: Anaconda.org 

含义是 GPU版本的cuda版本为11.3,安装的命令行语句如下(记得关闭魔法梯子,否则会下载不成功)

然后运行demo发现torch也被卸载掉了,那就继续重装:

进入torch找老版本,我的cuda 版本11.4 ,这个版本比较特殊,直接看作11.3即可

因此,torch 可以下载 

建议使用pip版本的,因为conda版本的命令我试过,没下载成功,因为开不开魔法梯子我都无法在命令行访问到anaconda官网,离谱,,明明刚还用conda下载了dglcu

pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

下载完成后再重新运行一下 ,运行下dgl的示例demo : 

import dgl
import torch as th
u, v = th.tensor([0, 1, 2]), th.tensor([2, 3, 4])
g = dgl.graph((u, v))
g.ndata['x'] = th.randn(5, 3)   # 原始特征在CPU上
print(g.device)
cuda_g = g.to('cuda:0')         # 接受来自后端框架的任何设备对象
cuda_g.device
cuda_g.ndata['x'].device        # 特征数据也拷贝到了GPU上
# 由GPU张量构造的图也在GPU上
u, v = u.to('cuda:0'), v.to('cuda:0')
g = dgl.graph((u, v))
print(g.device)

 然后就ok

http://www.lryc.cn/news/230397.html

相关文章:

  • 回文数和复利的威力(C#)
  • 【Java】面向对象程序设计 课程笔记 Java核心类
  • 16个值得推荐的.NET ORM框架
  • Git 进阶使用
  • 【微软技术栈】C#.NET 泛型数学
  • 【nlp】1.1文本处理的基本方法
  • 流量分析(信息安全铁人三项赛分区赛2-5.18)
  • 云服务器如何选?腾讯云2核2G3M云服务器88元一年!
  • 【Hello Go】初识Go语言
  • 计算机视觉:人脸识别与检测
  • 【NLP】理解 Llama2:KV 缓存、分组查询注意力、旋转嵌入等
  • ctyunos 与 openeuler
  • 跟着GPT学设计模式之工厂模式
  • VScode+python开发,多个解释器切换问题
  • c++ 经典服务器开源项目Tinywebserver如何运行
  • c++之xml的创建,增删改查
  • 【前端开发】JS Vue React中的通用递归函数
  • 【python 生成器 面试必备】yield关键字,协程必知必会系列文章--自己控制程序调度,体验做上帝的感觉 1
  • 头哥实践平台之MapReduce基础实战
  • Linux基础知识——tmux和vim
  • Java Web——TomcatWeb服务器
  • Zookeeper 命令使用和数据说明
  • 索尼RSV文件怎么恢复为MP4视频
  • pytorch-gpu(Anaconda3+cuda+cudnn)
  • 解析数据洁净之道:BI中数据清理对见解的深远影响
  • efcore反向共工程,单元测试
  • 利用IP风险画像强化金融行业网络安全防御
  • 1334. 阈值距离内邻居最少的城市
  • Live800:客服行业的发展历程及未来前景
  • exsi的安装和配置