sketch_simplification
sketch_simplification copied to clipboard
bobbens
GPU size error?
when I execute this command: python simplify.py the error as follows: THCudaCheck FAIL file=/pytorch/torch/lib/THC/generic/THCStorage.cu line=58 error=2 : out of memory
My GPU is 1080TI 11G, can you tell me how much computing resources are required for your model? Thanks!
Dependent on input image resolution size, but with 11G you should be able to handle fairly large images of at least 1024x1024 although likely quite a bit more. Make sure you have latest pytorch/cuda/cudnn (cudnn makes a big difference in my experience).
The same problem with nvidia K1000
Processing fig06_danshi.png...THCudaCheck FAIL file=/pytorch/torch/lib/THC/generic/THCStorage.c line=82 error=2 : out of memory ./figs.sh: línea 3: 22867 Violación de segmento python simplify.py --img "$IN" --out "$OUT"
image size is 170x224
@bobbens Thanks! I have run it on my MACBOOK PRO without CUDA，and the python version is 3.6.maybe the reason is python version is 3.6 not 2.7.It's a very amazing job,but the effect is still not ideal. Do you have a plan to release your code of your network structure? I want to do some work on your basis.
The same problem with nvidia K5200 THCudaCheck FAIL file=/opt/conda/conda-bld/pytorch_1503966894950/work/torch/lib/THC/generic/THCStorage.c line=79 error=2 : out of memory What's your pytorch/cuda/cudnn version? @bobbens Could you please release your whole code of your project, including the discriminator? Thank you so much!
@YgRen the network structure is actually included in the .t7 file. Most of my tests have been done using Cuda 9.1, Cudnn 7.5, and a GTX 1080. Using older cards will probably hit memory issues as apparently the torch7 code is much more memory intensive. When I get some time I'll try converting to native pytorch which should be more efficient.
i have using Cuda 9.1, Cudnn 7.5, and a GTX 1080，but when i run all the demo cmd， i have the problem “out of memory” plz help me
GTX 1050 Ti gives the same error (RuntimeError: CUDA error: out of memory), someone solve it?
