一、首先下载jaxlib
需要去这个非官方网站去找到适合自己的版本,下载到本地
然后使用对应的虚拟环境pip install 该文件名
即可
二、然后下载对应的jax
一行命令即可
pip install jax[cuda111]==0.3.7 -f https://whls.blob.core.windows.net/unstable/index.html --use-deprecated legacy-resolver
需要去这个非官方网站去找到适合自己的版本,下载到本地
然后使用对应的虚拟环境pip install 该文件名
即可
一行命令即可
pip install jax[cuda111]==0.3.7 -f https://whls.blob.core.windows.net/unstable/index.html --use-deprecated legacy-resolver