e3nn环境安装

一、创建环境

conda create -n e3nn python=3.7
y
conda activate e3nn

二、安装库

安装依赖库(我感觉先安装numpy等库之后在安装pytorch会特别快，不理解为什么)

conda install numpy==1.19.5
y
conda install matplotlib
y
conda install -c conda-forge scipy
y
conda install -c conda-forge scikit-learn
y
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
y

先安装torch_geometric

python -m pip install --upgrade pip
pip install torch_geometric