l2p论文环境安装(2) 复刻源码低版本环境版

从这个镜像网址可以看到所有jax发行版和支持的cuda：https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

安装过程

首先，源代码的环境依赖如requirements.txt)所示。要求环境自带cuda10，因为依赖jaxlib==0.1.68版本特别低，对应cuda也很旧。

从官网给的镜像jax_cuda_releases.html找到符合要求的安装包，比如笔者是cuda10.2，python3.7，就下载jaxlib-0.1.68+cuda102-cp37-none-manylinux2010_x86_64.whl，然后安装它：

pip install jaxlib-0.1.68+cuda102-cp37-none-manylinux2010_x86_64.whl

安装jax和flax

pip install jax==0.2.17
pip install flax==0.3.5

然后安装其它依赖即可。