知识嵌入消息传递神经网络KEMPNN 实验实现
接前文 1. 环境 查询cuda版本 >> nvcc -V nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2019 NVIDIA Corporation Built on Sun_Jul_28_19:12:52_Pacific_Daylight_Time_2019 Cuda compilation tools, release 10.1, V10.1.243 安装环境和包 conda create –name GNN python=3.8 conda activate GNN pip install torch==1.8.1+cu101 torchvision==0.9.1+cu101 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html conda remove -n GNN –all 在如下环境中测试通过。 Python 3.9, torch 1.9.1, cuda11.1 Python 3.8, torch 1.7.1, cuda10.1 2...