环境:Ubuntu 20.04 +pytorchGPU版本