1、先创建pytorch环境:conda create -n ljj_torch112 python=3.8
看本机的:
先看自己的cuda版本:(最权威的看:nvcc --version)
10.0的cuda于是不太符合,所以换一个10.2的cuda比较常用!
创建pytorch环境:
(资料:(258条消息) Ubuntu中安装Pytorch_ubuntu安装pytorch_不写八个的博客-CSDN博客)
2、下载pytorch(带上了cuda10.2)
我要下载的是1.12的,不是2.0的新版,所以到这里去找以前的pytorch:
Start Locally | PyTorch
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=10.2 -c pytorch
2.1 查看torch、cuda等的版本
(258条消息) 在Python中,查看torch、cuda和cudnn的版本_cudnn版本查看_初识-CV的博客-CSDN博客
3、下载DGL
Deep Graph Library (dgl.ai)
conda install -c dglteam/label/cu102 dgl
3.1 可以用conda list查看DGL是否下载成功