在Windows上安装CUDA12.1版本的DGL库

官网有pip和conda安装命令，我用pip只能安装上cpu版本，因此使用cuda命令安装

# If you have installed dgl-cudaXX.X package, please uninstall it first.
pip install  dgl -f https://data.dgl.ai/wheels/cu121/repo.html
conda install -c dglteam/label/cu121 dgl

会提示有很多包需要更新，输入y同意即可

检查是否安装成功：

import dgl
import torch

# 检查 DGL 和 CUDA 版本
print(f'DGL version: {dgl.__version__}')
print(f'CUDA available: {torch.cuda.is_available()}')
print(f'CUDA version: {torch.version.cuda}')

输出dgl的版本是2.0.0+cu121，说明CUDA12.1版本的DGL安装成功。

# 创建一个简单的图，并检查是否可以使用 CUDA
u = torch.tensor([0, 1, 2])
v &#