Polish CUDA Information stdout (#29109)

5 years ago · 7ae3cb554a
parent 2a864c70c4
commit 7ae3cb554a
1 changed files with 3 additions and 1 deletions
--- a/paddle/fluid/platform/device_context.cc
+++ b/paddle/fluid/platform/device_context.cc
@ -318,7 +318,9 @@ CUDADeviceContext::CUDADeviceContext(CUDAPlace place) : place_(place) {
  runtime_version_ = GetCUDARuntimeVersion(place_.device);

  LOG_FIRST_N(WARNING, 1) << "Please NOTE: device: " << place_.device
-                          << ", CUDA Capability: " << compute_capability_
+                          << ", GPU Compute Capability: "
+                          << compute_capability_ / 10 << "."
+                          << compute_capability_ % 10
                          << ", Driver API Version: " << driver_version_ / 1000
                          << "." << (driver_version_ % 100) / 10
                          << ", Runtime API Version: "