add gpu memory profiling, add single gpu option

add gpu memory profiling, add single gpu option #459