Allocation of 67108864 exceeds 10% of system memory

[superwang11]

Hello, what is the minimum computer configuration required to train the network?

[Jing-lun]

You're using CPU to train this network, try to install tensorflow-gpu==1.13 with CUDA 10.0 then this issue would be gone

[elhamravanbakhsh]

I installed TensorFlow-gpu ,but still, the code is running on CPU. I get the system memory error! Help please!

[ybhhdt]

I installed TensorFlow-gpu ,but still, the code is running on CPU. I get the system memory error! Help please!
I also installed tensorflow-gpu==1.15, but it still remind me 'exceeds 10% of system memory'. Did tensorflow need some code to load the model into GPU such as 'to(params.device)' in the Pytorch?