error: Assertion failure: prealloc_size <= total_mem when I try to train the Lego

[Zdmai]

RTX3050 mobile, Ubuntu 22.04.2 LTS x86_64

I don't know how to solve this problem, I try to change the patch_size into 2048 even 512 in train_nsvf_lego.sh and train.py but it can't work for me.

What can I do next.

./scripts/train_nsvf_lego.sh
[Taichi] version 1.7.0, llvm 15.0.4, commit 40ba02a3, linux, python 3.10.6
[Taichi] Starting on arch=cuda
[E 04/20/23 00:28:48.932 11717] [llvm_runtime_executor.cpp:materialize_runtime@570] Assertion failure: prealloc_size <= total_mem

Traceback (most recent call last):
File "/home/zzz/lec/taichi-nerfs/train.py", line 279, in
taichi_init(hparams)
File "/home/zzz/lec/taichi-nerfs/train.py", line 273, in taichi_init
ti.init(**taichi_init_args)
File "/home/zzz/.local/lib/python3.10/site-packages/taichi/lang/misc.py", line 455, in init
impl.get_runtime().prog.materialize_runtime()
RuntimeError: [llvm_runtime_executor.cpp:materialize_runtime@570] Assertion failure: prealloc_size <= total_mem

[bobcao3]

You would need to find the ti.init statement and then lower the memory allocation number.

[Zdmai]

yes! I solve this (last night, but my roommate have to sleep so I go to sleep too) by modify the

device_memory_GB to 1.0 that is work

but it have another error for me!