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

2024-02-20 09:30

本文主要是介绍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

win上重新安装 :anaconda + python + pytorch + CUDA也不知道需不需要安装 ,,,

anaconda的安装有两个重要的点:

一个是选择使用人的时候选择:all user :

其次是记得检查一下环境变量,如果anaconda没有个你自动添加上,记得anaconda安装完添加在系统变量的path路径里面:

==》因此,在下载anaconda的时候,截图保存下,到底吧anaconda下载到哪里了,如下图所示,别到时候找不到anaconda的一些Bin啊,Script啊,啥的 环境变量的位置

下载anaconda的参考:Anaconda超详细安装教程(Windows环境下)_conda安装-CSDN博客


由于我们的安装顺序是 :安装pycharm,此时pycharm中没有python环境,

然后我们安装anaconda,anaconda一般会给你创建一个 root 的环境

现在我们需要将pycharm使用anaconda创建的环境,就是那个初始的 root 环境 :方法是

右键选中Interpreter 设置,然后就可以在 add new interpreter 中选择 anaconda环境了:

选择使用现存的anaconda环境,anaconda一般就可以自己在你的电脑上匹配出来对应的位置了

点击OK,就可以使用这个anaconda环境了,加载这个环境中包含的库需一段时间耐心等待一下吧

下载 torch :如果有GPU记得下载GPU版本的torch,CPU和GPU版本的torch好像不兼容,所以注意下

pytorch下载地址:https://pytorch.org/

按照自己的电脑环境,获得下载的命令行语句:

然后在pycharm这个IDLE里面直接使用terminal,把这条语句打进去:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

之所以使用pycharm自己的terminal是为了确保在我们想要安装的anaconda环境中(现在的anaconda的环境只有最初始的Base环境一个,以后多了容易安装错,所以在pycharm的terminal中方便) 

确保安装的位置没错

 安装完了就完事啦

最后运行下述代码,确保GPU版本的Torch安装成功:

import torch# 检查是否有可用的GPU
if torch.cuda.is_available():device = torch.device("cuda")  # 使用GPUprint("GPU is available")
else:device = torch.device("cpu")  # 使用CPUprint("GPU is not available, using CPU")# 创建一个随机张量并将其移动到设备上
x = torch.rand(3, 3).to(device)# 打印设备类型和张量
print("Device:", device)
print("Tensor:", x)

然后可能出现 “import torch” 语句的 torch 部分红色波浪线,报错显示没找到这个库

没关系,是因为虽然下载成功Torch库,IDLE也需要花一段时间加载告知IDLE,你直接运行他是会报错的,重启一遍pycharm就没事了

终于,安装DGL:

这篇关于dgl 的cuda 版本 环境配置(dgl cuda 版本库无法使用问题解决)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/727754

相关文章

Zookeeper安装和配置说明

一、Zookeeper的搭建方式 Zookeeper安装方式有三种,单机模式和集群模式以及伪集群模式。 ■ 单机模式:Zookeeper只运行在一台服务器上,适合测试环境; ■ 伪集群模式:就是在一台物理机上运行多个Zookeeper 实例; ■ 集群模式:Zookeeper运行于一个集群上,适合生产环境,这个计算机集群被称为一个“集合体”(ensemble) Zookeeper通过复制来实现

CentOS7安装配置mysql5.7 tar免安装版

一、CentOS7.4系统自带mariadb # 查看系统自带的Mariadb[root@localhost~]# rpm -qa|grep mariadbmariadb-libs-5.5.44-2.el7.centos.x86_64# 卸载系统自带的Mariadb[root@localhost ~]# rpm -e --nodeps mariadb-libs-5.5.44-2.el7

中文分词jieba库的使用与实景应用(一)

知识星球:https://articles.zsxq.com/id_fxvgc803qmr2.html 目录 一.定义: 精确模式(默认模式): 全模式: 搜索引擎模式: paddle 模式(基于深度学习的分词模式): 二 自定义词典 三.文本解析   调整词出现的频率 四. 关键词提取 A. 基于TF-IDF算法的关键词提取 B. 基于TextRank算法的关键词提取

使用SecondaryNameNode恢复NameNode的数据

1)需求: NameNode进程挂了并且存储的数据也丢失了,如何恢复NameNode 此种方式恢复的数据可能存在小部分数据的丢失。 2)故障模拟 (1)kill -9 NameNode进程 [lytfly@hadoop102 current]$ kill -9 19886 (2)删除NameNode存储的数据(/opt/module/hadoop-3.1.4/data/tmp/dfs/na

hadoop开启回收站配置

开启回收站功能,可以将删除的文件在不超时的情况下,恢复原数据,起到防止误删除、备份等作用。 开启回收站功能参数说明 (1)默认值fs.trash.interval = 0,0表示禁用回收站;其他值表示设置文件的存活时间。 (2)默认值fs.trash.checkpoint.interval = 0,检查回收站的间隔时间。如果该值为0,则该值设置和fs.trash.interval的参数值相等。

NameNode内存生产配置

Hadoop2.x 系列,配置 NameNode 内存 NameNode 内存默认 2000m ,如果服务器内存 4G , NameNode 内存可以配置 3g 。在 hadoop-env.sh 文件中配置如下。 HADOOP_NAMENODE_OPTS=-Xmx3072m Hadoop3.x 系列,配置 Nam

Hadoop数据压缩使用介绍

一、压缩原则 (1)运算密集型的Job,少用压缩 (2)IO密集型的Job,多用压缩 二、压缩算法比较 三、压缩位置选择 四、压缩参数配置 1)为了支持多种压缩/解压缩算法,Hadoop引入了编码/解码器 2)要在Hadoop中启用压缩,可以配置如下参数

Makefile简明使用教程

文章目录 规则makefile文件的基本语法:加在命令前的特殊符号:.PHONY伪目标: Makefilev1 直观写法v2 加上中间过程v3 伪目标v4 变量 make 选项-f-n-C Make 是一种流行的构建工具,常用于将源代码转换成可执行文件或者其他形式的输出文件(如库文件、文档等)。Make 可以自动化地执行编译、链接等一系列操作。 规则 makefile文件

好题——hdu2522(小数问题:求1/n的第一个循环节)

好喜欢这题,第一次做小数问题,一开始真心没思路,然后参考了网上的一些资料。 知识点***********************************无限不循环小数即无理数,不能写作两整数之比*****************************(一开始没想到,小学没学好) 此题1/n肯定是一个有限循环小数,了解这些后就能做此题了。 按照除法的机制,用一个函数表示出来就可以了,代码如下

hdu1043(八数码问题,广搜 + hash(实现状态压缩) )

利用康拓展开将一个排列映射成一个自然数,然后就变成了普通的广搜题。 #include<iostream>#include<algorithm>#include<string>#include<stack>#include<queue>#include<map>#include<stdio.h>#include<stdlib.h>#include<ctype.h>#inclu