(feature) Update gpu_stress_test.py

[dmitrygalanov]

There is different hardware and it would be great to support dynamic memory depending on the type of GPU in the clusters.

Features: Dynamically retrieve the amount of memory from gpu device