conda pytorch-geomentric使用笔记

查看torch版本

cuda和cudnn下载

https://developer.nvidia.com/cuda-toolkit-archive

https://developer.nvidia.com/rdp/cudnn-archive

查看pytorch使用的cuda版本号

python -c "import torch; print(torch.version.cuda)"

export CUDA_HOME=/home/feng/.local/conda/

加个国内的镜像源，直接上命令：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

升级conda

conda update -n base -c defaults conda

conda查看已安装的软件

conda list --show-channel-urls

conda安装pytorch

conda install -c anaconda tensorflow-gpu

PIP

pip install --no-cache-dir torch-geometric