torch-geometric:一个图神经网络的工具包。
零。安装步骤
官方链接:https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html
1,新建conda环境(可选)
conda create -n geo1 python=3.6
conda activate geo1
conda deactivate
nvcc --version
10.0
nvcc 版本为10.0 ,所有想找torch 100的,但Geometric似乎只支持torch1.4和torch1.5。
2, 安装pytorch和对应驱动
pytorch官网
pip install torch==1.4.0+cu100-f https://download.pytorch.org/whl/torch_stable.html
注意,这个cu100非常重要,一不小心就白下载并且安装不了。
完成后查看版本
python -c "import torch; print(torch.version.cuda)"
python -c "import torch; print(torch.__version__)"
torch.version.cuda 10.0
torch.version 1.4.0+cu100
到这里,已经完成最重要的一步了,就是nvcc --version
和torch.version.cuda
的版本均为10.0,这里要相同。后面的问题记录都是因为这两者不同引起的。
3,清空cache(最好这样)
因为Stored in directory: /root/.cache/pip,所以
rm -rf ~/.cache/pip
网上还有一种方法,
pip install --verbose -