清华镜像地址

pip install xxx -i https://pypi.tuna.tsinghua.edu.cn/simple

安装pytorch

# CUDA 11.8
conda install pytorch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 pytorch-cuda=11.8 -c pytorch -c nvidia

安装DGL

#If you have installed dgl-cudaXX.X package, please uninstall it first.
conda install -c dglteam/label/cu118 dgl

导入 DGL 并检查版本以验证安装是否成功：

import dgl
print(dgl.__version__)

查看python版本

import sys
print(sys.version)

3.10.12 (main, Nov 20 2023, 15:14:05) [GCC 11.4.0]

查看pytorch版本

import torch
print(torch.__version__)

2.2.1+cu121

查看dgl版本命令

import dgl
print(dgl.__version__)

查看cuda版本的命令

import torchprint(torch.version.cuda)

查看GPU

import torchprint(torch.__version__)  # 查看torch当前版本号print(torch.cuda.device_count()) # 查看可用的CUDA数量print(torch.version.cuda)  # 编译当前版本的torch使用的cuda版本号print(torch.cuda.is_available())  # 查看GPU是否可以使用

nvidia-smi

NVIDIA-SMI 537.34                
Driver Version: 537.34       
CUDA Version: 12.2