首先查看cuda版本:
bash
nvcc --version
确定为12.4后:
安装pytorch
bash
conda install pytorch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 pytorch-cuda=12.4 -c pytorch -c nvidia
(其他版本查看:Previous PyTorch Versions | PyTorch )
安装DGL:
bash
pip install dgl -f https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
(其他版本查看:Deep Graph Library )