/
deepgcn_env_install.sh
44 lines (35 loc) · 1.67 KB
/
deepgcn_env_install.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
#!/usr/bin/env bash
# make sure command is : source deepgcn_env_install.sh
# install anaconda3.
# cd ~/
# wget https://repo.anaconda.com/archive/Anaconda3-2019.07-Linux-x86_64.sh
# bash Anaconda3-2019.07-Linux-x86_64.sh
export TORCH_CUDA_ARCH_LIST="7.0;7.5" # v100: 7.0; 2080ti: 7.5; titan xp: 6.1
# make sure system cuda version is the same with pytorch cuda
# follow the instruction of PyTorch Geometric: https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html
export PATH=/usr/local/cuda-10.2/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH
#conda create -n deepgcn
#conda activate deepgcn
# make sure pytorch version >=1.4.0
conda install -y pytorch=1.7.1 torchvision cudatoolkit=10.2 python=3.8 -c pytorch
pip install tensorboard
# command to install pytorch geometric, please refer to the official website for latest installation.
# https://pytorch-geometric.readthedocs.io/en/latest/notes/installation.html
CUDA=cu102
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.7.0+${CUDA}.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.7.0+${CUDA}.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.7.0+${CUDA}.html
pip install torch-geometric
pip install requests
# install useful modules
pip install tqdm
# additional environment required by deepergcn
pip install ogb
### check the version of ogb installed, if it is not the latest
python -c "import ogb; print(ogb.__version__)"
# please update the version by running
pip install -U ogb
pip install svgpathtools
pip install opencv-python