Zainstaluj TensorFlow za pomocą pip

Linuksa

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])))"

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

Windows WSL2

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

Procesor

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

Nocny

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

Linuksa

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. Konfiguracja GPU

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

Zainstaluj sterownik procesora graficznego NVIDIA, jeśli jeszcze tego nie zrobiłeś. Możesz użyć następującego polecenia, aby sprawdzić, czy jest ono zainstalowane.

nvidia-smi

3. Utwórz środowisko wirtualne za pomocą Venv

Moduł venv jest częścią standardowej biblioteki Pythona i jest oficjalnie zalecanym sposobem tworzenia środowisk wirtualnych.

Przejdź do żądanego katalogu środowisk wirtualnych i utwórz nowe środowisko Venv o nazwie tf za pomocą następującego polecenia.

python3 -m venv tf 

Możesz go aktywować za pomocą następującego polecenia.

source tf/bin/activate    

Upewnij się, że środowisko wirtualne jest aktywowane do końca instalacji.

4. 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.

# For GPU users
pip install tensorflow[and-cuda]
# For CPU users
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 zwrócony zostanie tensor, 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. Jeśli nie, przejdź do następnego kroku .

6. [Tylko GPU] Konfiguracja środowiska wirtualnego

Jeśli test GPU opisany w ostatniej sekcji zakończył się niepowodzeniem, najbardziej prawdopodobną przyczyną jest to, że komponenty nie są wykrywane i/lub powodują konflikt z istniejącą instalacją CUDA systemu. Aby to naprawić, musisz dodać kilka dowiązań symbolicznych.

• Utwórz dowiązania symboliczne do udostępnionych bibliotek NVIDIA:

pushd $(dirname $(python -c 'print(__import__("tensorflow").__file__)'))
ln -svf ../nvidia/*/lib/*.so* .
popd

• Utwórz dowiązanie symboliczne do 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

Sprawdź konfigurację GPU:

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

MacOS

1. Wymagania systemowe

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

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

2. Sprawdź wersję Pythona

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

python3 --version
python3 -m pip --version

3. 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

4. Sprawdź instalację

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

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

Natywny system 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, z tego pakietu wymagany jest plik msvcp140_1.dll (który może nie być dostarczany ze starszych pakietów redystrybucyjnych). Wersja redystrybucyjna jest dostarczana z programem Visual Studio 2019, ale można ją zainstalować osobno:

1. Przejdź do plików do pobrania 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 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ą procesora graficznego. Tworzy oddzielne środowisko, aby uniknąć zmiany zainstalowanego oprogramowania w systemie. Jest to również najłatwiejszy sposób zainstalowania wymaganego oprogramowania, zwłaszcza do konfiguracji procesora graficznego.

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 on aktywowany do końca instalacji.

5. Konfiguracja GPU

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

Najpierw zainstaluj sterownik procesora graficznego NVIDIA, jeśli jeszcze tego nie zrobiłeś.

Następnie zainstaluj CUDA, cuDNN z Condą.

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:

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

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

Sprawdź konfigurację GPU:

python -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 wersji systemu Windows 10 21H2, aktualizacji z listopada 2021 r.

Zobacz następujące dokumenty, aby:

• Pobierz najnowszą aktualizację systemu Windows 10 .
• Zainstaluj WSL2
• Skonfiguruj obsługę procesora graficznego NVIDIA® w WSL2

2. Konfiguracja GPU

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

Zainstaluj sterownik procesora graficznego NVIDIA, jeśli jeszcze tego nie zrobiłeś. Możesz użyć następującego polecenia, aby sprawdzić, czy jest ono zainstalowane.

nvidia-smi

3. 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.

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

4. Sprawdź instalację

Sprawdź konfigurację procesora:

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

Jeśli zwrócony zostanie tensor, 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.