• 一个全新的ubuntu20台式机，在Additional Drivers安装nvidia-470-server（一开始安装450，cunda版本只能到11.0，torch有些库用不了，可以直接切换点击Apply Changes重启就行）
  

• nvidia-smi查看CUDA Version可到11.4，手动安装11.2，以下地址下载并安装，安装完成后加入环境变量，用nvcc -V测试是否安装成功

https://developer.nvidia.com/cuda-11.2.0-download-archive?sudo sh cuda_11.2.0_460.27.04_linux.run// 配置cuda环境变量
gedit ~/.bashrc
export PATH=/usr/local/cuda-11.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
source ~/.bashrc

• 根据torch-geometric选择合适的pytorch版本，torch-1.9.0+cu111，装完torch几个版本库之后再装torch_geometric，指定版本2.0.4别太高了，安装完成后使用conda list torch查看安装情况

https://pytorch-geometric.com/whl/下载torch-1.9.0+cu111-cp38-cp38-linux_x86_64.whl手动安装
下载地址https://download.pytorch.org/whl/torch_stable.html
ctrl+f 查找cu111
wget https://download.pytorch.org/whl/cu111/torch-1.9.0%2Bcu111-cp38-cp38-linux_x86_64.whlpip install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-1.9.0+cu111.html
pip install --no-index torch-sparse -f https://pytorch-geometric.com/whl/torch-1.9.0+cu111.html
pip install --no-index torch-cluster -f https://pytorch-geometric.com/whl/torch-1.9.0+cu111.html
pip install --no-index torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.9.0+cu111.htmlpip install torch_geometric==2.0.4 -i https://pypi.doubanio.com/simple

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

---

• 安装JAX和cuda11.0时一样
• 之前安装cuda11.0时候查看pytorch对应版本conda list torch

pytorch                   1.7.1           py3.8_cuda11.0.221_cudnn8.0.5_0    pytorch

• 以下网址选择对应Jax版本，cuda11，cudnn选择和pytorch一样的805

https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

wget https://storage.googleapis.com/jax-releases/cuda11/jaxlib-0.3.18+cuda11.cudnn805-cp38-cp38-manylinux2014_x86_64.whl
pip install jaxlib-0.3.18+cuda11.cudnn805-cp38-cp38-manylinux2014_x86_64.whl

• 对应安装jax0.3.18
  pip install jax==0.3.18
• 打印GPU测试一下是否安装成功
• 版本匹配很重要，匹配对应版本

---

• 参考可能出现的问题
  JAX: 库安装和GPU使用，解决不能识别gpu问题https://blog.csdn.net/Papageno_Xue/article/details/125754893
• whl is not a supported wheel on this platform，查看conda是否激活版本，python版本是否为3.8
• 不行小则删除环境conda remove，大则重装系统安装驱动
• Ubuntu20.04下CUDA、cuDNN的详细安装与配置过程（图文）https://blog.csdn.net/weixin_37926734/article/details/123033286
• 类似运行报错： /usr/lib64/libstdc++.so.6: version GLIBCXX_3.4.21 not found 的问题，不用升级GCC，知乎上有讲将scipy版本降级或者在.bashrc文件添加conda的lib export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$HOME/miniconda/lib