תוכן עניינים:

כיצד אוכל לגרום ל-TensorFlow להשתמש ב-GPU שלי?
כיצד אוכל לגרום ל-TensorFlow להשתמש ב-GPU שלי?

וִידֵאוֹ: כיצד אוכל לגרום ל-TensorFlow להשתמש ב-GPU שלי?

וִידֵאוֹ: כיצד אוכל לגרום ל-TensorFlow להשתמש ב-GPU שלי?
וִידֵאוֹ: Matt Rocklin: Data Science, Dask, Scale | Turn the Lens #12 2024, מאי
Anonim

שלבים:

  1. הסר את ההתקנה שֶׁלְךָ ישן זרימת טנסור .
  2. להתקין זרימת טנסור - gpu התקנת pip זרימת טנסור - gpu .
  3. התקן את Nvidia כרטיס מסך & דרייברים (כנראה שכבר יש לך)
  4. הורד והתקן את CUDA.
  5. הורד והתקן את cuDNN.
  6. אמת על ידי תוכנית פשוטה.

יתר על כן, האם TensorFlow צריך GPU?

TensorFlow GPU תמיכה דורש מבחר של נהגים וספריות. הגדרה זו בלבד דורש ה-NVIDIA® GPU נהגים. הוראות התקנה אלו מיועדות למהדורה האחרונה של TensorFlow . ראה את תצורות ה-build שנבדקו עבור גרסאות CUDA ו-cuDNN לשימוש עם ישנות יותר TensorFlow משחרר.

באופן דומה, כיצד אוכל להשתמש במספר GPUs ב- TensorFlow? אם TensorFlow לפעולה יש גם מעבד וגם GPU יישומים, TensorFlow יבצע אוטומטית את הפעולה עַל א GPU המכשיר תחילה. אם יש לך יותר מאחד GPU , ה GPU עם המזהה הנמוך ביותר ייבחר כברירת מחדל. למרות זאת, TensorFlow לא מכניס פעולות לתוך מספר GPUs באופן אוטומטי.

בהתחשב בכך, מהו GPU ב-TensorFlow?

א GPU (יחידת עיבוד גרפית) הוא רכיב של רוב המחשבים המודרניים שנועד לבצע חישובים הדרושים לגרפיקה תלת מימדית.

האם TensorFlow יכול לפעול על AMD GPU?

הקוד הזה יכול לרוץ באופן טבעי על AMD כמו גם Nvidia GPU . כן אפשר להפעיל tensorflow עַל GPU של AMD אבל זו תהיה בעיה עצומה. כפי ש זרימת טנסור משתמש ב-CUDA שהוא קנייני פחית לא לרוץ על GPU של AMD אז אתה צריך להשתמש ב-OPENCL בשביל זה ו זרימת טנסור לא כתוב בזה.

מוּמלָץ: