Hi, all,
I implement my code about an Autoendocer with dense blcok on the basis of Pytorch. The total number of parameters around 0.8M. The image size is 100*100. the batchsize for training 100. However, when I launch CUDA, an error "cuda out of memory" will be ocurred. Can anyone can help me. My GPU card :
-----------------------------------------------------------------------------+
| NVIDIA-SMI 430.14 Driver Version: 430.14 CUDA Version: 10.2 |
|-------------------------------±---------------------±---------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 745 Off | 00000000:01:00.0 On | N/A |
| 25% 59C P0 N/A / N/A | 4037MiB / 4043MiB | 1% Default |
±------------------------------±---------------------±---------------------+
±----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|===========================================================================
Thanks.