安装torch以及torch-geometric的心酸

1、torch分为CPU版和GPU版,官网安装链接Start Locally | PyTorch

(1)如图,安装CPU版,根据操作系统OS、语言环境Language、计算平台Compute Platform的真实情况选择,CPU版的Compute Platform选择CPU即可,复制官网给出的安装命令即可。

安装torch以及torch-geometric的心酸_第1张图片

(2)安装GPU版,首先要查看电脑上的CUDA版本,在NVIDIA控制面板-系统信息-组件中可以看到,我的在更新前,CUDA版本是11.6,因此在选择Compute Platform时,按照上一张图片内容进行的选择,复制代码如下,也可以在选择过去的版本 Previous PyTorch Versions | PyTorch选择更为匹配的版本。

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

安装torch以及torch-geometric的心酸_第2张图片

 2、安装torch-geometric

按照anaconda下安装torch_geometric_F_aF_a的博客-CSDN博客_anaconda 安装 torch_geometric

安装即可。

pytorch的版本
确定pytorch的版本,最低1.4.0

python -c "import torch; print(torch.__version__)"
>>> 1.12.0

CUDA的版本
确定cuda的版本

python -c "import torch; print(torch.version.cuda)"
>>> 11.6

安装包
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric

将其中${TORCH}和${CUDA}替换为第一二步确定的torch和cuda版本。

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu102.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.8.0+cu102.html
pip install torch-geometric
————————————————
以上部分内容未转载,原文链接:https://blog.csdn.net/qq_37252519/article/details/119379232

3.如果安装过程中报错,可在网站 https://pytorch geometric.com选择与torch对应的版本,下载whl文件。

4.尤其要注意的一点时,安装torch-geometric时,一定要选择对应torch和cuda的版本,跟本机一致才能成功运行,否则会出现以下错误:

OSError: [WinError 127] 找不到指定的程序。 Error loading "D:\soft\anaconda\envs\pytorch\Lib\site-packages\torch\libcaffe2_detectron_ops.dll’ ,or one of its dependencies
 

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