You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
When I used tf.LoadSavedModel to load tensorflow models saved in python and used Session.Run to perform training, I found that gpu memory allocation was very high. But I didn't have this problem when I ran the training in python. I guess it's because tensorflow pre-allocates gpu memory in tfgo.
my question is
My question is how should I disable pre-allocation in tfgo to achieve the following effect in python?
When I used tf.LoadSavedModel to load tensorflow models saved in python and used Session.Run to perform training, I found that gpu memory allocation was very high. But I didn't have this problem when I ran the training in python. I guess it's because tensorflow pre-allocates gpu memory in tfgo.
my question is
My question is how should I disable pre-allocation in tfgo to achieve the following effect in python?
Or setting like tf.ConfigProto()
The text was updated successfully, but these errors were encountered: