JAX
文章平均质量分 64
cccrick
这个作者很懒,什么都没留下…
展开
-
CUDA安装、排错
该文档介绍了为什么/usr/local下会有多个前缀为cuda的目录以及如何安装和维护多个版本的CUDA。此外,还介绍了可能出现的一些排错情况和如何解决。例如,当系统加载了CuDNN 8.1的运行时库,但代码是使用CuDNN 8.6编译的时,可能会出现版本不匹配的错误。为了解决这个问题,可以更新运行时库,重新编译源代码,或回退到使用8.1版本的cuDNN。总之,建议在开发环境中使用相同版本的运行时库和源代码进行开发和测试,以避免版本不匹配引起的问题。原创 2023-03-23 21:48:57 · 1020 阅读 · 0 评论 -
为什么 JAX 数组是不可变的,而 NumPy 数组不是呢?
本文介绍了 JAX 数组的不可变性是出于设计的考虑,因为 JAX 建立在函数式编程的概念上,函数不会修改它们的输入,而是创建新的对象作为输出。所有的数组变换都是实现为纯函数,它们接受一个输入数组并返回一个新的数组。这意味着在 JAX 数组上执行的任何操作都将创建一个新的数组,而原始数组保持不变。相比之下,NumPy 数组默认是可变的,这意味着您可以直接修改 NumPy 数组的内容,而不创建新的数组。文章指出,JAX 的设计可以更容易地理解程序的行为,并避免由可变状态导致的常见错误。然而,如果需要就地修改数组原创 2023-03-23 17:02:49 · 208 阅读 · 0 评论