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

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

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

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

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

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

conda install numba

یا

pip install numba

• قدم دوم: اجرای دستورهای زیر در محیط پایتون

>> from numba import cuda
>> cuda.detect()

• تمام!

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

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

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