如何查看实际占用显存大小?

taichi在使用gpu并行时,在初始化时会通过device_memory_fraction之类的分配显存,在整个运行期间显存占用都不会变化,我想知道的能不能看到实际占用了多少显存?

Taichi在ti.init(arch=ti.cuda)时有一个预分配制度,这部分预分配的显存暂时还不支持查看。不过我们在即将发布的1.7.0里面把大部分的分配改为了动态分配,预分配的显存显著减少,所以届时nvidia-smi看到的显存占用就是taichi所用到的显存大小。

你可以安装taichi-nightly来提前使用这个feature:

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