How to allocate special amount of GPU memory


#1

Is there anyway to allocate special amount of GPU memory to the program via pytorch? in other words, can I block the memory while running my code?


(Simon Wang) #2

What do you mean by blocking the memory?


#3

I mean blocking the memory in a way that that amount would be set for your program to be used.
Meaning: e.g. you block X amount of gpu memory so X amount of GPU memory would be allocated for your code usage and if another person use the gpu they cannot use it. meaning you block X amount of GPU memory