首先查看cuda版本:

nvcc --version

确定为12.4后:

安装pytorch

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:

pip install  dgl -f https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html

(其他版本查看:Deep Graph Library )

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐