TensorFlow را با پیپ نصب کنید

این راهنما برای آخرین نسخه پایدار TensorFlow است. برای ساخت پیش نمایش (شبیه) ، از بسته پیپ به نام tf-nightly استفاده کنید. برای نیازهای نسخه قدیمی تر TensorFlow به این جداول مراجعه کنید. برای ساخت فقط CPU، از بسته پیپ با نام tensorflow-cpu استفاده کنید.

در اینجا نسخه های سریع دستورات نصب آمده است. برای دستورالعمل های گام به گام به پایین اسکرول کنید.

لینوکس

python3 -m pip install 'tensorflow[and-cuda]'
# Verify the installation:
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

MacOS

# There is currently no official GPU support for MacOS.
python3 -m pip install tensorflow
# Verify the installation:
python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

ویندوز بومی

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
# Anything above 2.10 is not supported on the GPU on Windows Native
python -m pip install "tensorflow<2.11"
# Verify the installation:
python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

ویندوز WSL2

python3 -m pip install tensorflow[and-cuda]
# Verify the installation:
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

CPU

python3 -m pip install tensorflow
# Verify the installation:
python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

شبانه

python3 -m pip install tf-nightly
# Verify the installation:
python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

الزامات سخت افزاری

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

  • کارت گرافیک NVIDIA® با معماری CUDA® 3.5، 5.0، 6.0، 7.0، 7.5، 8.0 و بالاتر. لیست کارت‌های GPU با قابلیت CUDA® را ببینید.
  • برای پردازنده‌های گرافیکی با معماری‌های CUDA® پشتیبانی‌نشده، یا برای جلوگیری از کامپایل JIT از PTX، یا استفاده از نسخه‌های مختلف کتابخانه‌های NVIDIA®، به راهنمای ساخت Linux از منبع مراجعه کنید.
  • بسته ها به جز آخرین معماری CUDA پشتیبانی شده، حاوی کد PTX نیستند. بنابراین، وقتی CUDA_FORCE_PTX_JIT=1 تنظیم شده باشد، TensorFlow روی GPU های قدیمی بارگذاری نمی شود. (برای جزئیات بیشتر به سازگاری برنامه مراجعه کنید.)

سیستم مورد نیاز

  • اوبونتو 16.04 یا بالاتر (64 بیتی)
  • macOS 12.0 (Monterey) یا بالاتر (64 بیتی) (بدون پشتیبانی از GPU)
  • Windows Native - Windows 7 یا بالاتر (64 بیتی) (بعد از TF 2.10 پشتیبانی از GPU وجود ندارد)
  • Windows WSL2 - Windows 10 19044 یا بالاتر (64 بیت)

الزامات نرم افزاری

نرم افزار NVIDIA® زیر فقط برای پشتیبانی GPU مورد نیاز است.

دستورالعمل های گام به گام

لینوکس

1. سیستم مورد نیاز

  • اوبونتو 16.04 یا بالاتر (64 بیتی)

TensorFlow فقط به طور رسمی از اوبونتو پشتیبانی می کند. با این حال، دستورالعمل‌های زیر ممکن است برای سایر توزیع‌های لینوکس نیز کار کنند.

2. راه اندازی GPU

اگر فقط TensorFlow را روی CPU اجرا می کنید، می توانید از این بخش رد شوید.

اگر نصب نکرده اید درایور کارت گرافیک NVIDIA را نصب کنید. می توانید از دستور زیر برای بررسی نصب بودن آن استفاده کنید.

nvidia-smi

3. یک محیط مجازی با venv ایجاد کنید

ماژول venv بخشی از کتابخانه استاندارد پایتون است و راه رسمی توصیه شده برای ایجاد محیط های مجازی است.

به دایرکتوری محیط های مجازی مورد نظر خود بروید و با دستور زیر یک محیط venv جدید به نام tf ایجاد کنید.

python3 -m venv tf 

با دستور زیر می توانید آن را فعال کنید.

source tf/bin/activate    

مطمئن شوید که محیط مجازی برای بقیه مراحل نصب فعال است.

4. TensorFlow را نصب کنید

TensorFlow به نسخه اخیر پیپ نیاز دارد، بنابراین نصب پیپ خود را ارتقا دهید تا مطمئن شوید که آخرین نسخه را اجرا می کنید.

pip install --upgrade pip

سپس تنسورفلو را با پیپ نصب کنید.

# For GPU users
pip install tensorflow[and-cuda]
# For CPU users
pip install tensorflow

6. نصب را بررسی کنید

بررسی تنظیمات CPU:

python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

اگر یک تانسور برگردانده شود، شما TensorFlow را با موفقیت نصب کرده اید.

تنظیم GPU را تأیید کنید:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

اگر لیستی از دستگاه های GPU برگردانده شود، TensorFlow را با موفقیت نصب کرده اید. اگر نه به مرحله بعد ادامه دهید .

6. [فقط GPU] پیکربندی محیط مجازی

اگر آزمایش GPU در بخش آخر ناموفق بود، محتمل‌ترین علت این است که اجزا شناسایی نمی‌شوند و/یا با نصب CUDA سیستم موجود تداخل دارند. بنابراین باید چند پیوند نمادین برای رفع این مشکل اضافه کنید.

  • پیوندهای نمادین به کتابخانه های مشترک NVIDIA ایجاد کنید:
pushd $(dirname $(python -c 'print(__import__("tensorflow").__file__)'))
ln -svf ../nvidia/*/lib/*.so* .
popd
  • یک پیوند نمادین به ptxas ایجاد کنید:
ln -sf $(find $(dirname $(dirname $(python -c "import nvidia.cuda_nvcc;         
print(nvidia.cuda_nvcc.__file__)"))/*/bin/) -name ptxas -print -quit) $VIRTUAL_ENV/bin/ptxas

تنظیم GPU را تأیید کنید:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

MacOS

1. سیستم مورد نیاز

  • macOS 10.12.6 (Sierra) یا بالاتر (64 بیتی)

در حال حاضر هیچ پشتیبانی رسمی GPU برای اجرای TensorFlow در MacOS وجود ندارد. دستورالعمل های زیر برای اجرا بر روی CPU هستند.

2. نسخه پایتون را بررسی کنید

بررسی کنید که آیا محیط پایتون شما قبلاً پیکربندی شده است:

python3 --version
python3 -m pip --version

3. TensorFlow را نصب کنید

TensorFlow به نسخه اخیر پیپ نیاز دارد، بنابراین نصب پیپ خود را ارتقا دهید تا مطمئن شوید که آخرین نسخه را اجرا می کنید.

pip install --upgrade pip

سپس تنسورفلو را با پیپ نصب کنید.

pip install tensorflow

4. نصب را بررسی کنید

python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

اگر یک تانسور برگردانده شود، شما TensorFlow را با موفقیت نصب کرده اید.

ویندوز بومی

1. سیستم مورد نیاز

  • ویندوز 7 یا بالاتر (64 بیت)

2. Microsoft Visual C++ Redistributable را نصب کنید

Microsoft Visual C++ Redistributable را برای Visual Studio 2015، 2017 و 2019 نصب کنید. با شروع نسخه TensorFlow 2.1.0، فایل msvcp140_1.dll از این بسته مورد نیاز است (که ممکن است از بسته های قابل توزیع مجدد قدیمی ارائه نشود). قابل توزیع مجدد با Visual Studio 2019 ارائه می شود اما می توان آن را به طور جداگانه نصب کرد:

  1. به دانلودهای Microsoft Visual C++ بروید.
  2. صفحه را به سمت پایین اسکرول کنید و به بخش Visual Studio 2015، 2017 و 2019 بروید.
  3. Microsoft Visual C++ Redistributable را برای Visual Studio 2015، 2017 و 2019 برای پلتفرم خود دانلود و نصب کنید.

مطمئن شوید که مسیرهای طولانی در ویندوز فعال هستند .

3. Miniconda را نصب کنید

Miniconda رویکرد توصیه شده برای نصب TensorFlow با پشتیبانی از GPU است. این یک محیط مجزا برای جلوگیری از تغییر هر نرم افزار نصب شده در سیستم شما ایجاد می کند. این نیز ساده ترین راه برای نصب نرم افزار مورد نیاز مخصوصاً برای راه اندازی GPU است.

Miniconda Windows Installer را دانلود کنید. روی فایل دانلود شده دوبار کلیک کنید و دستورالعمل های روی صفحه را دنبال کنید.

4. یک محیط کوندا ایجاد کنید

با دستور زیر یک محیط conda جدید به نام tf ایجاد کنید.

conda create --name tf python=3.9

با دستورات زیر می توانید آن را غیرفعال و فعال کنید.

conda deactivate
conda activate tf

مطمئن شوید که برای بقیه مراحل نصب فعال است.

5. راه اندازی GPU

اگر فقط TensorFlow را روی CPU اجرا می کنید، می توانید از این بخش رد شوید.

اگر نصب نکرده اید ابتدا درایور کارت گرافیک NVIDIA را نصب کنید.

سپس CUDA، cuDNN را با conda نصب کنید.

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0

6. TensorFlow را نصب کنید

TensorFlow به نسخه اخیر پیپ نیاز دارد، بنابراین نصب پیپ خود را ارتقا دهید تا مطمئن شوید که آخرین نسخه را اجرا می کنید.

pip install --upgrade pip

سپس تنسورفلو را با پیپ نصب کنید.

# Anything above 2.10 is not supported on the GPU on Windows Native
pip install "tensorflow<2.11" 

7. نصب را بررسی کنید

بررسی تنظیمات CPU:

python -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

اگر یک تانسور برگردانده شود، شما TensorFlow را با موفقیت نصب کرده اید.

تنظیم GPU را تأیید کنید:

python -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

اگر لیستی از دستگاه های GPU برگردانده شود، TensorFlow را با موفقیت نصب کرده اید.

ویندوز WSL2

1. سیستم مورد نیاز

  • ویندوز 10 19044 یا بالاتر (64 بیت). این مربوط به ویندوز 10 نسخه 21H2، به‌روزرسانی نوامبر 2021 است.

به اسناد زیر مراجعه کنید:

2. راه اندازی GPU

اگر فقط TensorFlow را روی CPU اجرا می کنید، می توانید از این بخش رد شوید.

اگر نصب نکرده اید درایور کارت گرافیک NVIDIA را نصب کنید. می توانید از دستور زیر برای بررسی نصب بودن آن استفاده کنید.

nvidia-smi

3. TensorFlow را نصب کنید

TensorFlow به نسخه اخیر پیپ نیاز دارد، بنابراین نصب پیپ خود را ارتقا دهید تا مطمئن شوید که آخرین نسخه را اجرا می کنید.

pip install --upgrade pip

سپس تنسورفلو را با پیپ نصب کنید.

# For GPU users
pip install tensorflow[and-cuda]
# For CPU users
pip install tensorflow

4. نصب را بررسی کنید

بررسی تنظیمات CPU:

python3 -c "import tensorflow as tf; print(tf.reduce_sum(tf.random.normal([1000, 1000])))"

اگر یک تانسور برگردانده شود، شما TensorFlow را با موفقیت نصب کرده اید.

تنظیم GPU را تأیید کنید:

python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

اگر لیستی از دستگاه های GPU برگردانده شود، TensorFlow را با موفقیت نصب کرده اید.

محل بسته بندی

چند مکانیسم نصب نیاز به URL بسته TensorFlow Python دارند. مقداری که مشخص می کنید به نسخه پایتون شما بستگی دارد.

نسخه URL
لینوکس x86
پشتیبانی از GPU پایتون 3.9 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Python 3.9 فقط با CPU https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow_cpu-2.18.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
پشتیبانی از GPU پایتون 3.10 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Python 3.10 فقط با CPU https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow_cpu-2.18.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
پشتیبانی از GPU پایتون 3.11 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Python 3.11 فقط برای CPU https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow_cpu-2.18.0-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
پشتیبانی از GPU پایتون 3.12 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Python 3.12 فقط برای CPU https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow_cpu-2.18.0-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Linux Arm64 (فقط CPU)
پایتون 3.9 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
پایتون 3.10 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
پایتون 3.11 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
پایتون 3.12 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
macOS x86 (فقط CPU)
احتیاط : TensorFlow 2.16 آخرین نسخه TensorFlow بود که از macOS x86 پشتیبانی می کرد
پایتون 3.9 https://storage.googleapis.com/tensorflow/versions/2.16.2/tensorflow-2.16.2-cp39-cp39-macosx_10_15_x86_64.whl
پایتون 3.10 https://storage.googleapis.com/tensorflow/versions/2.16.2/tensorflow-2.16.2-cp310-cp310-macosx_10_15_x86_64.whl
پایتون 3.11 https://storage.googleapis.com/tensorflow/versions/2.16.2/tensorflow-2.16.2-cp311-cp311-macosx_10_15_x86_64.whl
پایتون 3.12 https://storage.googleapis.com/tensorflow/versions/2.16.2/tensorflow-2.16.2-cp312-cp312-macosx_10_15_x86_64.whl
macOS Arm64 (فقط CPU)
پایتون 3.9 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp39-cp39-macosx_12_0_arm64.whl
پایتون 3.10 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp310-cp310-macosx_12_0_arm64.whl
پایتون 3.11 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp311-cp311-macosx_12_0_arm64.whl
پایتون 3.12 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp312-cp312-macosx_12_0_arm64.whl
ویندوز (فقط CPU)
پایتون 3.9 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp39-cp39-win_amd64.whl
پایتون 3.10 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp310-cp310-win_amd64.whl
پایتون 3.11 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp311-cp311-win_amd64.whl
پایتون 3.12 https://storage.googleapis.com/tensorflow/versions/2.18.0/tensorflow-2.18.0-cp312-cp312-win_amd64.whl