How to specify GPU memory fraction usage in pytorch?

1

As in TensorFlow we can specify GPU memory fraction(as given below) , how can we do the same in Pytorch?

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
1 Like
2

Hi,

There is no such option in pytorch. It will allocate the memory as it needs it.

4 Likes