查看“Jax”的源代码
←
Jax
跳到导航
跳到搜索
因为以下原因,您没有权限编辑本页:
您请求的操作仅限属于该用户组的用户执行:
用户
您可以查看和复制此页面的源代码。
*简单的说就是GPU加速、支持自动微分(autodiff)的numpy。参考[https://jax.readthedocs.io/en/latest/notebooks/quickstart.html] ==安装== *jax和jaxlib版本要匹配(注意cuda支持) *我的CUDA版本是11.1 ,卡是A40,系统是Ubuntu,cudnn版本是805 *安装最新版本 jax 0.2.26和jaxlib0.1.75后会在random函数报错“CustomCall failed: jaxlib/cuda_prng_kernels.cc:30: operation cudaGetLastError() failed: the provided PTX was compiled with an unsupported toolchain” *最后安装的 jax0.2.1 (pip install -v jax=0.2.1), jaxlib是0.1.71 [https://storage.googleapis.com/jax-releases/cuda111/jaxlib-0.1.71+cuda111-cp38-none-manylinux2010_x86_64.whl]解决问题 ]
返回至“
Jax
”。
导航菜单
个人工具
登录
命名空间
页面
讨论
大陆简体
已展开
已折叠
查看
阅读
查看源代码
查看历史
更多
已展开
已折叠
搜索
导航
首页
社群首页
最近更改
随机页面
帮助
工具
链入页面
相关更改
特殊页面
页面信息