问答网站开发,住房和城乡规划建设局网站,wordpress 多域名绑定,腾讯云备案网站建设方案书1. 如果你同时有dgl dglcu-XX.XX 那么#xff0c;应该只会运行dgl #xff08;DGL的CPU版本#xff09;#xff0c;因此#xff0c;你需要把dgl(CPU)版本给卸载了
但是我只卸载CPU版本还不够#xff0c;我GPU 版本的dglcu依旧不好使#xff0c;因此吧GPU版本的也得卸载…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 torch1.11.0cu113 torchvision0.12.0cu113 torchaudio0.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
