1、pytorch 1.8.1 + CUDA11.1 对应的DGL版本是0.6.1(linux、windows也适用) 2、CUDA+pytorch+DGL安装 3、安装dgl-cuda 4、torch 安装备忘录 dgl文件下载 https://anaconda.org/dglteam/repo 1. 可以将下面网址中的cu111改成你想要的版本,比如cu113、cu117等 https://pypi.tuna.tsinghua.edu.cn/simple...
# 更新conda conda update conda # 创建一个新的虚拟环境(可选) conda create -n pytorch_env python=3.x #将3.x替换为你需要的Python版本 conda activate pytorch_env # 安装与CUDA 11.4兼容的PyTorch(注意版本号) conda install pytorch torchvision torchaudio cudatoolkit=11.4 -c pytorch 注意:由于conda的...
建议去下面链接下载后本地安装: https://download.pytorch.org/whl/torch/https://download.pytorch.org/whl/torchvision 查看torch和torchvision的版本匹配信息: https://pytorch.org/get-started/previous-versions/