一、首先下载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
参考:
《JAX windows 下的安装》
《Windows下 Jax安装(2022.4)》
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)