查询当前pytorch安装的对应cuda版本号
·
由于自己笔记本cuda是11.1版本，安装图网络学习框架DGL总是环境问题报错，如：
gl._ffi.base.DGLError: [10:54:05] C:\Users\Administrator\dgl\src\runtime\c_runtime_api.cc:87: Check failed: allow_missin
g: Device API gpu is not enabled. Please install the cuda version of dgl.
发现DGL包需要跟环境中pytorch安装cuda版本一致，因此记录。
查询当前pytorch对应cuda版本号，在终端输入：
python
import torch
print(torch.version.cuda)
欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
8
0- 0
已为社区贡献1条内容
所有评论(0)