请问sparse matrix有考虑加入CG, 并且在cuda后端支持float64吗

Otis · 2023 年4 月 12 日 02:00

houkensjtu · 2023 年4 月 12 日 07:29

CUDA 后端的 CG 已经有了一个版本可以使用的，请见：https://github.com/taichi-dev/taichi/pull/7035
暂时只支持 32 位，预计以后也会加入 64 位的支持。
另外，也有不是基于 SparseMatrix 的 Matrix-free CG 求解器在开发中，可以看这里的 RFC 和这个 PR

Otis · 2023 年4 月 12 日 12:57

请问LinearOperator需要用什么版本taichi

houkensjtu · 2023 年4 月 13 日 02:24

现在 LinearOperator 的 PR 已经合入主仓但是还没有在 Release 版本里（下一个版本 1.6.0 发布可能会到 5 月以后）。如果要使用的话你可以选择源码编译 Taichi，或者安装 taichi-nightly：

pip install -i https://pypi.taichi.graphics/simple/ taichi-nightly

注意安装多版本 Taichi 的时候最好用虚拟环境把他们隔离开避免交叉影响，有具体问题可以再私信交流。