search

bmild / nerf

Code release for NeRF (Neural Radiance Fields)
http://tancik.com/nerf
MIT License
9.97k stars 1.39k forks source link

How much video memory is needed to run the test example? #31

Closed 123Ray321 closed 4 years ago

123Ray321 commented 4 years ago

My laptop's video card is 1660Ti, and I always run out of resources. Runtime exception： tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[4194304,90] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc [Op:ConcatV2]name: concat

jaggzh commented 3 years ago

@123Ray321 Did you find a solution? I face the same issue. Decreasing the batch size, in the config file, from 1024, down to 16, still results in the same OOM error with that same tensor shape. I have a 4gb GTX970. https://github.com/bmild/nerf/issues/81