search

DeepInsight-PCALab / CompactBilinearPooling-Pytorch

A Pytorch Implementation for Compact Bilinear Pooling.
182 stars 41 forks source link

out of cuda memory #5

Closed pangjh3 closed 6 years ago

pangjh3 commented 6 years ago

I just run the test, and it run out of my cuda memory, which is 20G. Have I miss something?

pangjh3 commented 6 years ago

OK, problem fixed. My mistake setting on VISIBLE CUDA

dichen-cd commented 6 years ago

I was just trying to find an available GPU for memory test. Glad you fixed it.