选择cuda12.cudnn89 对应自己需要版本,根据自己要求选择,笔者选择为0.4.27,则执行: pip install jax==0.4.27 # 此处jaxlib笔者为whl本地安装 pip install jaxlib-0.4.27+cuda12.cudnn89-cp39-cp39-manylinux2014_x86_64.whl 切记笔者 chex==0.1.86,chex 对 jax 也有要求,容易遇到 bug ,全是泪与恨。
wgethttps://developer.download.nvidia.com/compute/cuda/11.2.1/local_installers/cuda_11.2.1_460.32.03_linux.runsudo 下载完成后进行安装: sh cuda_11.2.1_460.32.03_linux.run 机器如果已经安装过驱动,在安装cuda11.2时候如果勾选了driver就会报错,所以不用不勾选driver,只安装工具包。但是要注意,如果driver...