نشناختن GPU در tensorflow

Question

سلام.

من درایو gpu را نصب کردم. با استفاده از کد زیر لیست دستگاههای قابل استفاده بر روی سیستم را نمایش دادم فقط cpu را نشان میده و خبری از gpu نیست.

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

 نمایش خروجی:

2023-01-25 15:58:46.111772: I tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 269631437
locality {
}
incarnation: 37015345003685307405
]

Process finished with exit code 0

 

Answer 1

کدی که ارائه کردید دستگاه‌های موجود از طریق کتابخانه دستگاه تنسورفلو را بررسی می‌کند، که فقط دستگاه‌هایی را فهرست می‌کند که TensorFlow می‌تواند استفاده کند. به نظر می رسد که درایور GPU شما توسط TensorFlow شناسایی نمی شود، به همین دلیل است که در خروجی لیست نشده است.

 

چند چیز وجود دارد که می تواند باعث این مشکل شود. یکی از دلایل احتمالی این است که درایور GPU شما به درستی نصب نشده است. احتمال دیگر این است که TensorFlow برای استفاده از GPU شما پیکربندی نشده است.

 

برای بررسی اینکه آیا GPU توسط سیستم شناسایی شده است، می توانید دستور زیر را اجرا کنید:

nvidia-smi

اگر GPU شناسایی شود و درایور به درستی نصب شده باشد، باید خروجی مشابه زیر را ببینید:

 

اگر GPU شناسایی نشود، ممکن است مشکل از نصب درایور یا پیکربندی سیستم باشد.

 

چیز دیگری که می توانید امتحان کنید این است که متغیر محیط خود را بررسی کنید، اگر CUDA_VISIBLE_DEVICES تنظیم شده است، آن را از حالت تنظیم خارج کنید.

 

اگر همچنان مشکل دارید، لطفاً اطلاعات بیشتری در مورد پیکربندی سیستم خود و مراحلی که برای نصب درایور GPU انجام دادید ارائه دهید.

اگر GPU شناسایی نشود، ممکن است مشکل از نصب درایور یا پیکربندی سیستم باشد.

 

چیز دیگری که می توانید امتحان کنید این است که متغیر محیط خود را بررسی کنید، اگر CUDA_VISIBLE_DEVICES تنظیم شده است، آن را از حالت تنظیم خارج کنید.

 

اگر همچنان مشکل دارید، لطفاً اطلاعات بیشتری در مورد پیکربندی سیستم خود و مراحلی که برای نصب درایور GPU انجام دادید ارائه دهید.