本文档介绍了如何安装PyTorch的CPU和GPU版本，以及如何根据CUDA版本选择合适的torch-geometric。首先，通过官方链接选择对应的操作系统、语言环境和计算平台安装PyTorch。对于GPU版，需要检查CUDA版本并安装匹配的PyTorch。接着，安装torch-geometric，确保torch和CUDA版本与安装包相匹配。若遇到错误，可从pytorch-geometric.com下载对应版本的whl文件。务必注意torch-geometric版本与PyTorch及CUDA的兼容性，否则可能导致运行错误。
1、torch分为CPU版和GPU版，官网安装链接Start Locally | PyTorch
（1）如图，安装CPU版，根据操作系统OS、语言环境Language、计算平台Compute Platform的真实情况选择，CPU版的Compute Platform选择CPU即可，复制官网给出的安装命令即可。
（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
最低0.47元/天 解锁文章
扫一扫
2998
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
