检查cuda版本
nvidia-smi

查看pytorch版本
pip list
conda list

版本过低
这里最好去pytorch的官网查看自己需要哪款pytorch

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
参考官网 Installation — pytorch_geometric 2.0.2 documentation
pip install torch-scatter -f https://data.pyg.org/whl/torch-${TORCH}+${

本文档介绍了如何检查CUDA版本和PyTorch版本,并根据官方指南选择合适的PyTorch版本来安装torch_geometric库。对于CUDA 10.1,推荐使用PyTorch 1.7.1。
最低0.47元/天 解锁文章
3300

被折叠的 条评论
为什么被折叠?



