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這個不裝會報錯,其他依賴會自動安裝
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章