使用清华镜像快速安装pytorch
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda install pytorch torchvision cudatoolkit=9.0 (版本号从https://pytorch.org/上,查找对应版本)
下面就可以快速安装了。
若果做GNN推荐是用DGL框架,新手友好。