Updated run script to use the correct GPU device