Attend the Women in ML Symposium on December 7 Register now

Zainstaluj TensorFlow z pip

Zadbaj o dobrą organizację dzięki kolekcji Zapisuj i kategoryzuj treści zgodnie ze swoimi preferencjami.

Ten przewodnik dotyczy najnowszej stabilnej wersji TensorFlow. W przypadku wersji zapoznawczej (nightly) użyj pakietu pip o nazwie tf-nightly . Zapoznaj się z tymi tabelami , aby zapoznać się z wymaganiami dotyczącymi starszych wersji TensorFlow. W przypadku kompilacji tylko dla procesora użyj pakietu pip o nazwie tensorflow-cpu .

Oto szybkie wersje poleceń instalacyjnych. Przewiń w dół, aby zobaczyć instrukcje krok po kroku.

Linuks

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/
python3 -m pip install tensorflow
# Verify install:
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

System operacyjny Mac

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

Natywny Windows

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
python3 -m pip install "tensorflow<2.11"
# Verify install:
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

Windows WSL2

conda install -c conda-forge cudatoolkit=11.2 cudnn=8.1.0
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/
python3 -m pip install tensorflow
# Verify install:
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

procesor

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

Nocny

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

Wymagania sprzętowe

Obsługiwane są następujące urządzenia obsługujące GPU:

  • Karta GPU NVIDIA® z architekturą CUDA® 3.5, 5.0, 6.0, 7.0, 7.5, 8.0 i nowszymi. Zobacz listę kart GPU obsługujących technologię CUDA® .
  • W przypadku procesorów graficznych z nieobsługiwaną architekturą CUDA® lub w celu uniknięcia kompilacji JIT z PTX lub korzystania z różnych wersji bibliotek NVIDIA® zapoznaj się z przewodnikiem dotyczącym kompilacji systemu Linux ze źródeł .
  • Pakiety nie zawierają kodu PTX, z wyjątkiem najnowszej obsługiwanej architektury CUDA®; dlatego TensorFlow nie ładuje się na starszych procesorach graficznych, gdy CUDA_FORCE_PTX_JIT=1 . (Szczegółowe informacje znajdują się w sekcji Zgodność aplikacji ).

Wymagania systemowe

  • Ubuntu 16.04 lub nowszy (64-bitowy)
  • macOS 10.12.6 (Sierra) lub nowszy (64-bitowy) (bez obsługi GPU)
  • Windows Native — Windows 7 lub nowszy (64-bitowy)
  • Windows WSL2 — Windows 10 19044 lub nowszy (64-bitowy)

Wymagania Systemowe

Następujące oprogramowanie NVIDIA® jest wymagane tylko do obsługi GPU.

Instrukcje krok po kroku

Linuks

1. Wymagania systemowe

  • Ubuntu 16.04 lub nowszy (64-bitowy)

TensorFlow oficjalnie obsługuje tylko Ubuntu. Jednak poniższe instrukcje mogą również działać w przypadku innych dystrybucji Linuksa.

2. Zainstaluj Minicondę

Miniconda to zalecane podejście do instalacji TensorFlow z obsługą GPU. Tworzy osobne środowisko, aby uniknąć zmiany jakiegokolwiek zainstalowanego oprogramowania w systemie. Jest to również najłatwiejszy sposób na zainstalowanie wymaganego oprogramowania, zwłaszcza w przypadku konfiguracji GPU.

Możesz użyć następującego polecenia, aby zainstalować Minicondę. Podczas instalacji może być konieczne naciśnięcie klawisza Enter i wpisanie „tak”.

curl https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -o Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh

Może być konieczne ponowne uruchomienie terminala lub source ~/.bashrc , aby włączyć polecenie conda . Użyj conda -V , aby sprawdzić, czy została pomyślnie zainstalowana.

3. Utwórz środowisko Conda

Utwórz nowe środowisko conda o nazwie tf za pomocą następującego polecenia.

conda create --name tf python=3.9

Możesz go dezaktywować i aktywować za pomocą następujących poleceń.

conda deactivate
conda activate tf

Upewnij się, że jest aktywna do końca instalacji.

4. Konfiguracja GPU

Możesz pominąć tę sekcję, jeśli uruchamiasz TensorFlow tylko na procesorze.

Najpierw zainstaluj sterownik GPU NVIDIA, jeśli go nie masz. Możesz użyć następującego polecenia, aby sprawdzić, czy jest zainstalowane.

nvidia-smi

Następnie zainstaluj CUDA i cuDNN z condą.

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

Skonfiguruj ścieżki systemowe. Możesz to zrobić za pomocą następującego polecenia za każdym razem, gdy uruchamiasz nowy terminal po aktywacji środowiska conda.

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/

Dla Twojej wygody zaleca się zautomatyzowanie go za pomocą następujących poleceń. Ścieżki systemowe zostaną automatycznie skonfigurowane po aktywowaniu tego środowiska Conda.

mkdir -p $CONDA_PREFIX/etc/conda/activate.d
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/' > $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh

5. Zainstaluj TensorFlow

TensorFlow wymaga najnowszej wersji pip, więc zaktualizuj instalację pip, aby mieć pewność, że korzystasz z najnowszej wersji.

pip install --upgrade pip

Następnie zainstaluj TensorFlow za pomocą pip.

pip install tensorflow

6. Sprawdź instalację

Sprawdź konfigurację procesora:

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

Jeśli tensor zostanie zwrócony, pomyślnie zainstalowałeś TensorFlow.

Sprawdź konfigurację GPU:

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

Jeśli zostanie zwrócona lista urządzeń GPU, pomyślnie zainstalowałeś TensorFlow.

System operacyjny Mac

1. Wymagania systemowe

  • macOS 10.12.6 (Sierra) lub nowszy (64-bitowy)

Obecnie nie ma oficjalnej obsługi GPU do uruchamiania TensorFlow w systemie MacOS. Poniższe instrukcje dotyczą uruchamiania na procesorze.

2. Sprawdź wersję Pythona

Sprawdź, czy Twoje środowisko Pythona jest już skonfigurowane:

python3 --version
python3 -m pip --version

2. Zainstaluj Minicondę

Miniconda to zalecane podejście do instalacji TensorFlow z obsługą GPU. Tworzy osobne środowisko, aby uniknąć zmiany jakiegokolwiek zainstalowanego oprogramowania w systemie. Jest to również najłatwiejszy sposób na zainstalowanie wymaganego oprogramowania, zwłaszcza w przypadku konfiguracji GPU.

curl https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-x86_64.sh -o Miniconda3-latest-MacOSX-x86_64.sh
bash Miniconda3-latest-MacOSX-x86_64.sh

Może być konieczne ponowne uruchomienie terminala lub source ~/.bashrc , aby włączyć polecenie conda . Użyj conda -V , aby sprawdzić, czy została pomyślnie zainstalowana.

4. Utwórz środowisko Conda

Utwórz nowe środowisko conda o nazwie tf za pomocą następującego polecenia.

conda create --name tf python=3.9

Możesz go dezaktywować i aktywować za pomocą następujących poleceń.

conda deactivate
conda activate tf

Upewnij się, że jest aktywna do końca instalacji.

5. Zainstaluj TensorFlow

TensorFlow wymaga najnowszej wersji pip, więc zaktualizuj instalację pip, aby mieć pewność, że korzystasz z najnowszej wersji.

pip install --upgrade pip

Następnie zainstaluj TensorFlow za pomocą pip.

pip install tensorflow

6. Sprawdź instalację

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

Jeśli tensor zostanie zwrócony, pomyślnie zainstalowałeś TensorFlow.

Natywny Windows

1. Wymagania systemowe

  • Windows 7 lub nowszy (64-bitowy)

2. Zainstaluj pakiet redystrybucyjny Microsoft Visual C++

Zainstaluj pakiet redystrybucyjny Microsoft Visual C++ dla programu Visual Studio 2015, 2017 i 2019 . Począwszy od wersji TensorFlow 2.1.0, plik msvcp140_1.dll jest wymagany z tego pakietu (który może nie być dostarczany ze starszych pakietów redystrybucyjnych). Pakiet redystrybucyjny jest dostarczany z programem Visual Studio 2019 , ale można go zainstalować osobno:

  1. Przejdź do pobierania programu Microsoft Visual C++ .
  2. Przewiń stronę w dół do sekcji Visual Studio 2015, 2017 i 2019 .
  3. Pobierz i zainstaluj pakiet redystrybucyjny Microsoft Visual C++ dla programu Visual Studio 2015, 2017 i 2019 dla swojej platformy.

Upewnij się, że w systemie Windows włączone są długie ścieżki .

3. Zainstaluj Minicondę

Miniconda to zalecane podejście do instalacji TensorFlow z obsługą GPU. Tworzy osobne środowisko, aby uniknąć zmiany jakiegokolwiek zainstalowanego oprogramowania w systemie. Jest to również najłatwiejszy sposób na zainstalowanie wymaganego oprogramowania, zwłaszcza w przypadku konfiguracji GPU.

Pobierz Instalator Windows Miniconda . Kliknij dwukrotnie pobrany plik i postępuj zgodnie z instrukcjami wyświetlanymi na ekranie.

4. Utwórz środowisko Conda

Utwórz nowe środowisko conda o nazwie tf za pomocą następującego polecenia.

conda create --name tf python=3.9

Możesz go dezaktywować i aktywować za pomocą następujących poleceń.

conda deactivate
conda activate tf

Upewnij się, że jest aktywna do końca instalacji.

5. Konfiguracja GPU

Możesz pominąć tę sekcję, jeśli uruchamiasz TensorFlow tylko na CPU.

Najpierw zainstaluj sterownik GPU NVIDIA, jeśli go nie masz.

Następnie zainstaluj CUDA, cuDNN z conda.

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

6. Zainstaluj TensorFlow

TensorFlow wymaga najnowszej wersji pip, więc zaktualizuj instalację pip, aby mieć pewność, że korzystasz z najnowszej wersji.

pip install --upgrade pip

Następnie zainstaluj TensorFlow za pomocą pip.

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

7. Sprawdź instalację

Sprawdź konfigurację procesora:

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

Jeśli tensor zostanie zwrócony, pomyślnie zainstalowałeś TensorFlow.

Sprawdź konfigurację GPU:

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

Jeśli zostanie zwrócona lista urządzeń GPU, pomyślnie zainstalowałeś TensorFlow.

Windows WSL2

1. Wymagania systemowe

  • Windows 10 19044 lub nowszy (64-bitowy). Odpowiada to Windows 10 w wersji 21H2, aktualizacji z listopada 2021 r.

Zobacz następujące dokumenty, aby:

2. Zainstaluj Minicondę

Miniconda to zalecane podejście do instalacji TensorFlow z obsługą GPU. Tworzy osobne środowisko, aby uniknąć zmiany jakiegokolwiek zainstalowanego oprogramowania w systemie. Jest to również najłatwiejszy sposób na zainstalowanie wymaganego oprogramowania, zwłaszcza w przypadku konfiguracji GPU.

Możesz użyć następującego polecenia, aby zainstalować Minicondę. Podczas instalacji może być konieczne naciśnięcie klawisza Enter i wpisanie „tak”.

curl https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -o Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh

Może być konieczne ponowne uruchomienie terminala lub source ~/.bashrc , aby włączyć polecenie conda . Użyj conda -V , aby sprawdzić, czy została pomyślnie zainstalowana.

3. Utwórz środowisko Conda

Utwórz nowe środowisko conda o nazwie tf za pomocą następującego polecenia.

conda create --name tf python=3.9

Możesz go dezaktywować i aktywować za pomocą następujących poleceń.

conda deactivate
conda activate tf

Upewnij się, że jest aktywna do końca instalacji.

4. Konfiguracja GPU

Możesz pominąć tę sekcję, jeśli uruchamiasz TensorFlow tylko na procesorze.

Najpierw zainstaluj sterownik GPU NVIDIA, jeśli go nie masz. Możesz użyć następującego polecenia, aby sprawdzić, czy jest zainstalowane.

nvidia-smi

Następnie zainstaluj CUDA i cuDNN z condą.

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

Skonfiguruj ścieżki systemowe. Możesz to zrobić za pomocą następującego polecenia za każdym razem, gdy uruchamiasz nowy terminal po aktywacji środowiska conda.

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/

Dla Twojej wygody zaleca się zautomatyzowanie go za pomocą następujących poleceń. Ścieżki systemowe zostaną automatycznie skonfigurowane po aktywowaniu tego środowiska Conda.

mkdir -p $CONDA_PREFIX/etc/conda/activate.d
echo 'export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:$CONDA_PREFIX/lib/' > $CONDA_PREFIX/etc/conda/activate.d/env_vars.sh

5. Zainstaluj TensorFlow

TensorFlow wymaga najnowszej wersji pip, więc zaktualizuj instalację pip, aby mieć pewność, że korzystasz z najnowszej wersji.

pip install --upgrade pip

Następnie zainstaluj TensorFlow za pomocą pip.

pip install tensorflow

6. Sprawdź instalację

Sprawdź konfigurację procesora:

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

Jeśli tensor zostanie zwrócony, pomyślnie zainstalowałeś TensorFlow.

Sprawdź konfigurację GPU:

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

Jeśli zostanie zwrócona lista urządzeń GPU, pomyślnie zainstalowałeś TensorFlow.

Lokalizacja paczki

Kilka mechanizmów instalacyjnych wymaga adresu URL pakietu TensorFlow Python. Podana wartość zależy od używanej wersji języka Python.

Wersja Adres URL
Linuks
Obsługa GPU Pythona 3.7 https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.11.0-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Python 3.7 tylko dla procesorów https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.11.0-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Obsługa GPU Pythona 3.8 https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.11.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Python 3.8 tylko dla procesorów https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.11.0-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Obsługa GPU Pythona 3.9 https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.11.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Python 3.9 tylko dla procesorów https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.11.0-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Obsługa GPU Pythona 3.10 https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-2.11.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Python 3.10 tylko dla procesorów https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow_cpu-2.11.0-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
macOS (tylko procesor)
Pythona 3.7 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.11.0-cp37-cp37m-macosx_10_14_x86_64.whl
Pythona 3.8 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.11.0-cp38-cp38-macosx_10_14_x86_64.whl
Pythona 3.9 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.11.0-cp39-cp39-macosx_10_14_x86_64.whl
Pythona 3.10 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-2.11.0-cp310-cp310-macosx_10_14_x86_64.whl
Okna
Obsługa GPU Pythona 3.7 https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.11.0-cp37-cp37m-win_amd64.whl
Python 3.7 tylko dla procesorów https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.11.0-cp37-cp37m-win_amd64.whl
Obsługa GPU Pythona 3.8 https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.11.0-cp38-cp38-win_amd64.whl
Python 3.8 tylko dla procesorów https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.11.0-cp38-cp38-win_amd64.whl
Obsługa GPU Pythona 3.9 https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.11.0-cp39-cp39-win_amd64.whl
Python 3.9 tylko dla procesorów https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.11.0-cp39-cp39-win_amd64.whl
Obsługa GPU Pythona 3.10 https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-2.11.0-cp310-cp310-win_amd64.whl
Python 3.10 tylko dla procesorów https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow_cpu-2.11.0-cp310-cp310-win_amd64.whl