查看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
网友评论