Ten przewodnik dotyczy najnowszej stabilnej wersji TensorFlow. W przypadku kompilacji podglądu (nightly) użyj pakietu pip o nazwie tf-nightly
. Zapoznaj się z tymi tabelami , aby zapoznać się z wymaganiami starszych wersji TensorFlow. W przypadku kompilacji obsługującej tylko procesor użyj pakietu pip o nazwie tensorflow-cpu
.
Oto krótkie wersje poleceń instalacyjnych. Przewiń w dół, aby zapoznać się z instrukcjami krok po kroku.
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])))"
Wymagania sprzętowe
Obsługiwane są następujące urządzenia obsługujące procesor graficzny:
- Karta graficzna NVIDIA® z architekturami CUDA® 3.5, 5.0, 6.0, 7.0, 7.5, 8.0 i nowszymi. Zobacz listę kart graficznych obsługujących CUDA® .
- W przypadku procesorów graficznych z nieobsługiwaną architekturą CUDA®, aby uniknąć kompilacji JIT z PTX lub aby użyć różnych wersji bibliotek NVIDIA®, zapoznaj się z przewodnikiem dotyczącym kompilacji systemu Linux ze źródła .
- Pakiety nie zawierają kodu PTX z wyjątkiem najnowszej obsługiwanej architektury CUDA®; dlatego TensorFlow nie ładuje się na starszych procesorach graficznych, gdy ustawiono
CUDA_FORCE_PTX_JIT=1
. (Szczegółowe informacje można znaleźć w części 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) (brak obsługi GPU po TF 2.10)
- Windows WSL2 — Windows 10 19044 lub nowszy (64-bitowy)
Wymagania dotyczące oprogramowania
- Python 3.9–3.12
- pip w wersji 19.0 lub nowszej dla systemu Linux (wymaga obsługi
manylinux2014
) i Windows. pip w wersji 20.3 lub nowszej dla systemu macOS. - Natywny system Windows wymaga pakietu redystrybucyjnego Microsoft Visual C++ dla programu Visual Studio 2015, 2017 i 2019
Następujące oprogramowanie NVIDIA® jest wymagane wyłącznie do obsługi procesora graficznego.
- Sterowniki procesora graficznego NVIDIA®
- >= 525.60.13 dla Linuksa
- >= 528,33 dla WSL w systemie Windows
- Zestaw narzędzi CUDA® 12.3 .
- cuDNN SDK 8.9.7 .
- (Opcjonalnie) TensorRT poprawiający opóźnienia i przepustowość wnioskowania.
Instrukcje krok po kroku
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:
- Przejdź do plików do pobrania Microsoft Visual C++ .
- Przewiń stronę w dół do sekcji Visual Studio 2015, 2017 i 2019 .
- 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.
Lokalizacja pakietu
Kilka mechanizmów instalacji wymaga adresu URL pakietu TensorFlow Python. Określona wartość zależy od wersji języka Python.
Ten przewodnik dotyczy najnowszej stabilnej wersji TensorFlow. W przypadku kompilacji podglądu (nightly) użyj pakietu pip o nazwie tf-nightly
. Zapoznaj się z tymi tabelami , aby zapoznać się z wymaganiami starszych wersji TensorFlow. W przypadku kompilacji obsługującej tylko procesor użyj pakietu pip o nazwie tensorflow-cpu
.
Oto krótkie wersje poleceń instalacyjnych. Przewiń w dół, aby zapoznać się z instrukcjami krok po kroku.
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])))"
Wymagania sprzętowe
Obsługiwane są następujące urządzenia obsługujące procesor graficzny:
- Karta graficzna NVIDIA® z architekturami CUDA® 3.5, 5.0, 6.0, 7.0, 7.5, 8.0 i nowszymi. Zobacz listę kart graficznych obsługujących CUDA® .
- W przypadku procesorów graficznych z nieobsługiwaną architekturą CUDA®, aby uniknąć kompilacji JIT z PTX lub aby użyć różnych wersji bibliotek NVIDIA®, zapoznaj się z przewodnikiem dotyczącym kompilacji systemu Linux ze źródła .
- Pakiety nie zawierają kodu PTX z wyjątkiem najnowszej obsługiwanej architektury CUDA®; dlatego TensorFlow nie ładuje się na starszych procesorach graficznych, gdy ustawiono
CUDA_FORCE_PTX_JIT=1
. (Szczegółowe informacje można znaleźć w części 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) (brak obsługi GPU po TF 2.10)
- Windows WSL2 — Windows 10 19044 lub nowszy (64-bitowy)
Wymagania dotyczące oprogramowania
- Python 3.9–3.12
- pip w wersji 19.0 lub nowszej dla systemu Linux (wymaga obsługi
manylinux2014
) i Windows. pip w wersji 20.3 lub nowszej dla systemu macOS. - Natywny system Windows wymaga pakietu redystrybucyjnego Microsoft Visual C++ dla programu Visual Studio 2015, 2017 i 2019
Następujące oprogramowanie NVIDIA® jest wymagane wyłącznie do obsługi procesora graficznego.
- Sterowniki procesora graficznego NVIDIA®
- >= 525.60.13 dla Linuksa
- >= 528,33 dla WSL w systemie Windows
- Zestaw narzędzi CUDA® 12.3 .
- cuDNN SDK 8.9.7 .
- (Opcjonalnie) TensorRT poprawiający opóźnienia i przepustowość wnioskowania.
Instrukcje krok po kroku
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:
- Przejdź do plików do pobrania Microsoft Visual C++ .
- Przewiń stronę w dół do sekcji Visual Studio 2015, 2017 i 2019 .
- 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.
Lokalizacja pakietu
Kilka mechanizmów instalacji wymaga adresu URL pakietu TensorFlow Python. Określona wartość zależy od wersji języka Python.