Hoe TensorFlow op Ubuntu 24.04 te installeren
TensorFlow is een krachtige open-source bibliotheek ontwikkeld door Google voor machine learning en deep learning-taken, die veel wordt gebruikt door onderzoekers, ontwikkelaars en datawetenschappers om machine learning-modellen te bouwen en te trainen.
Als u een Ubuntu-machine gebruikt en TensorFlow wilt instellen, leidt deze handleiding u door het proces van het installeren en gebruiken van TensorFlow op uw systeem.
Voorwaarden
Voordat u TensorFlow installeert, moet u ervoor zorgen dat uw Ubuntu-machine aan de volgende vereisten voldoet:
- Python: TensorFlow ondersteunt Python 3.7 tot 3.10.
- Pip: Pip is een pakketbeheerder voor Python.
- Hardware: Hoewel TensorFlow op CPU's kan draaien, kan het hebben van een GPU de training voor grote modellen aanzienlijk versnellen.
Stap 1: Installeer Python en Pip in Ubuntu
Voordat u nieuwe software installeert, is het een goed idee om uw pakketlijst bij te werken en geïnstalleerde pakketten bij te werken.
sudo apt update
sudo apt upgrade
Installeer vervolgens Python en pip, een pakketbeheerder voor Python die wordt gebruikt om Python-pakketten te installeren en te beheren.
sudo apt install python3
sudo apt install python3-pip
Eenmaal geïnstalleerd, kunt u de installatie van Python en pip verifiëren voordat u pakketten zoals TensorFlow installeert.
python3 --version
pip3 --version
Stap 2: Maak een virtuele omgeving in Ubuntu
Het maken van een virtuele omgeving is optioneel, maar wordt aanbevolen, omdat het helpt om uw TensorFlow-installatie geïsoleerd te houden van andere Python-projecten.
Maak een nieuwe map voor uw TensorFlow-project en navigeer ernaartoe:
mkdir tensorflow_project
cd tensorflow_project
Creëer een virtuele omgeving en activeer deze:
python3 -m venv tensorflow_env
source tensorflow_env/bin/activate
Uw terminalprompt zou nu de naam van de virtuele omgeving moeten tonen, wat aangeeft dat deze actief is.
Stap 3: Installeer TensorFlow in Ubuntu
Nu uw omgeving is ingesteld, kunt u de nieuwste stabiele versie van TensorFlow samen met de bijbehorende afhankelijkheden installeren met behulp van pip, de Python-pakketbeheerder.
pip install --upgrade pip
pip install tensorflow
Stap 4: Controleer TensorFlow in Ubuntu
Nadat de installatie is voltooid, is het een goed idee om te controleren of TensorFlow correct is geïnstalleerd door naar de Python-shell te gaan.
python3
Importeer TensorFlow en controleer de versie:
import tensorflow as tf
print(tf.__version__)
Als TensorFlow correct is geïnstalleerd, wordt het versienummer foutloos afgedrukt.
2.18.0
U kunt ook een eenvoudige test uitvoeren om er zeker van te zijn dat TensorFlow werkt.
hello = tf.constant('Hello, TensorFlow!')
print(hello)
Dit zou zoiets als moeten opleveren.
tf.Tensor(b'Hello, TensorFlow!', shape=(), dtype=string)
Stap 5: Installeer TensorFlow met GPU-ondersteuning (optioneel)
Als u een compatibele NVIDIA GPU hebt en deze wilt gebruiken voor snellere berekeningen, kunt u TensorFlow installeren met GPU-ondersteuning.
lspci | grep -i nvidia
sudo apt install nvidia-driver-535 -y
sudo reboot
Installeer vervolgens TensorFlow met GPU-ondersteuning.
pip install tensorflow-gpu
Controleer de installatie door te controleren of TensorFlow uw GPU detecteert:
import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))
Conclusie
U hebt TensorFlow met succes geïnstalleerd op Ubuntu 24.04. Of u nu werkt aan een eenvoudig machine learning-project of een complex deep learning-model, TensorFlow biedt de tools die u nodig hebt om aan de slag te gaan.
Vergeet niet om uw omgeving georganiseerd te houden door virtuele omgevingen te gebruiken en overweeg GPU-ondersteuning als u snellere berekeningen nodig hebt.