图卷积神经网络环境配置

tech2024-08-07  56

图卷积神经网络环境配置

1.首先需要装好Anaconda3

版本为Anaconda3-5.2.0-Linux-x86_64.sh, 各版本的Anaconda下载地址:https://repo.anaconda.com/archive/ 下载好对应的sh文件之后,命令行输入 bash ~/Downloads/Anaconda3-5.2.0-Linux-x86_64.sh

2.安装python

终端输入: conda install python=3.6

3.Pytorch安装

命令行输入: conda install pytorch torchvision cudatoolkit=10.1 -c pytorch 其中10.1是GPU的cuda版本,Linux查看cuda版本: cat /usr/local/cuda/version.txt

4.pip安装图卷积依赖包

pip install torch-geometric pip install torch-sparse pip install torch-scatter

最新回复(0)