نمایش مدل و امتیاز کارت گرافیک در پایتون

مشکلی که خیلیا برای راه اندازی بستر برنامه نویسی یادگیری عمیق دارن اینه که نمی‌دونن کارت گرافیکشون، قابلیت اجرای کدهای تنسورفلو رو داره یا نه. یکی از راه‌های فهمیدنش اینه که nvidia compute capability رو سرچ کنیم یا از لینک زیر، امتیاز کارت گرافیکمون رو ببینیم (فعلا سایت انویدیا، ایران رو تحریم کرده و باید با تغییر آی‌پی، لینک رو باز کنید)

https://developer.nvidia.com/cuda-gpus

که در زمان نوشتن این مطلب، (طبق لینک زیر که باز هم تحریمیم و نیاز به تغییر آی‌پی داره!) اجرای تنسورفلو روی GPU نیاز به امتیاز بالاتر از 3.5 داره

https://www.tensorflow.org/install/gpu

ولی مشکل اینجاست که توی این لیست، همه‌ی مدل‌های کارت گرافیک وجود نداره و احتمالش هست که کارت گرافیک ما هم توش نباشه.

خبر خوب اینه که یک راه ساده وجود داره که توی پایتون بتونیم امتیاز کارت گرافیکمون رو بفهمیم. در ادامه، این راه دوم رو با هم می‌بینیم:

  • قدم اول: نصب کتابخونه numba

conda install numba

یا

pip install numba

  • قدم دوم: اجرای دستورهای زیر

from numba import cuda

cuda.detect()

  • تمام!

مثلا برای لپ تاپ من که کارت گرافیک Geforce MX130 داره (و اتفاقا فعلا توی لیست سایت انویدیا نیست!) خروجی به شکل زیر بود:

numba.cuda.detect() output

که مدل کارت گرافیک رو نشون میده و می‌گه که امتیاز 5 داره و ساپورت می‌شه!! (در حالی که ظرفیت 2 گیگ داره و در نگاه اول شاید به نظر نمی‌رسید که اینطور باشه!)

البته من tensorflow-gpu رو هم نصب و تست کردم و ساپورتش تایید شد :thumbsup:

ارسال نظر