当前位置: 代码迷 >> 综合 >> 【win10】安装pytorch1.6.0+cuda10.0 + torch-geometric
  详细解决方案

【win10】安装pytorch1.6.0+cuda10.0 + torch-geometric

热度:81   发布时间:2024-02-20 03:05:31.0

安装pytorch1.6.0

pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

安装cuda10.0+cudnn

参考,
百度云链接:https://pan.baidu.com/s/1HqyGjha1IkS6vBFlzxQ82g
提取码:ls3d

安装torch-geometric

pip install torch-scatter==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-sparse==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-cluster==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-spline-conv==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
pip install torch-geometric
  相关解决方案