最近需要使用到pytorch-geometric,捣鼓了好久的环境,终于成功啦!
使用vscode远程访问服务器,并安装环境,环境如下:
Ubuntu: 20.04.4
GCC version: 9.4.0
Python: 3.6.10
CUDA: 10.1.243
1.创建环境并指定Python版本
conda create -n XXX pip python==3.6.10
2.安装Pytorch
根据系统在网站上(PyTorch)找到可安装的Pytorch版本,若需要其他版本,参看网址(Previous PyTorch Versions | PyTorch)。Start Locally | PyTorch
注意:pytorch的cuda版与服务器本地的cuda版本。
安装命令:
pip install torch==1.4.0 torchvision==0.5.0
conda install cudatoolkit=10.1 -c pytorch
3.安装pytorch-geometric
(1)pytorch-geometric官网:Installation — pytorch_geometric 2.0.5 documentation
(2)老版本pytorch对应的pytorch-geometric网址:https://pytorch-geometric.com/whl/
(3)以下命令中文件下载网址:https://pytorch-geometric.com/whl/torch-1.4.0.html
(4)安装命令:
pip install torch_scatter-latest+cu101-cp36-cp36m-linux_x86_64.whl
pip install torch_sparse-latest+cu101-cp36-cp36m-linux_x86_64.whl
pip install torch_spline_conv-latest+cu101-cp36-cp36m-linux_x86_64.whl
pip install torch-cluster==1.4.5
pip install torch_geometric==1.4.3
wow安装完成并运行成功!!!那就继续搬砖吧!哭唧唧!
最后,DGL库安装官网如下:
找到对应版本即可生成安装命令,高效实用