bianlongpeng
bianlongpeng
I have solved it by the code: 
you can turn down your batch size，may be 32 to 16，for me it soved this problem！
@hoainv99 I also meet this error, but for me the realy error is "include/THC/THCGeneral.h:12:10: fatal error: cuda.h: No such file or directory", this is becauese commond can't find cuda dir....
@JoinTyang Firstly, to get the depth image, you shuold reference YadiraF/PRNet Repository! There are writes.py and api.py. Secondly, set `depth_image = get_depth_image(vertices, prn.triangles, h, w, True)` to `depth_image = get_depth_image(vertices,...
@tomjeans thanks, it works