cuda 10 环境下安装 pytorch_geometric

cuda 10 环境下安装 pytorch_geometric

pip install torch==1.4.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html

pip install torch-scatter==latest+cu100 -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-sparse==latest+cu100 -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-cluster==latest+cu100 -f https://pytorch-geometric.com/whl/torch-1.4.0.html
pip install torch-spline-conv==latest+cu100 -f https://pytorch-geometric.com/whl/torch-1.4.0.html

pip install pytest-runner
pip install torch-geometric

注意事项:

  • 安装要点就是所有的软件包cuda版本要一样
  • 当前最新版为1.5.0,但是没有cuda10的版本,只有cuda10.1的版本
  • pip install pytest-runner这个不装会报错,其他依赖会自动安装

你可能感兴趣的:(环境搭建,pytorch,cuda)