Created by: barrierye
fix that the GPU version did not indicate the environment variable CUDA_VISIBLE_DEVICES