安装torch-geometric

安装torch-geometric(亲身体验)

1.首先安装torch,在安装torch之前,检查一下自己电脑的显卡驱动版本。(版本不符合会导致下载之后cuda驱动不起来)如我们在这里安装1.7.1PyTorch /1.1.0 CUDA版本,在官网查找命令https://anaconda.org/pytorch-lts/pytorch,还可以在此网站上找到对应版本用pip下载

pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple/

2.torch-geometric的安装和其他四个库相关(torch-cluster、pytorch-scatter、torch-sparse、pytorch-spline-conv)
在https://pytorch-geometric.com/whl/index.html中,我们顺着随便一个自己的torch版本进去看。安装命令如下

conda install -c esri torch-sparse
conda install -c esri torch-cluster
conda install -c rusty1s pytorch-scatter
conda install -c rusty1s pytorch-spline-conv

3.安装torch-geometric

pip install torch-geometric

在这里插入图片描述
安装成功!!!

你可能感兴趣的:(笔记,pytorch,深度学习,python)