TensorFlow var izmantot CPU un GPU, lai aprēķinātu sarežģītus mākslīgā intelekta (AI) un mašīnmācīšanās (ML) aprēķinus. TensorFlow var izmantot jebkuru CUDA atbalstītu NVIDIA GPU, lai paātrinātu AI/ML programmas. Ja jums nav CUDA atbalstīta GPU, TensorFlow izmantos CPU AI/ML kodiem. Bez GPU paātrinājuma TensorFlow veiktspēja pasliktināsies sarežģītās AI/ML programmās.
Šajā rakstā mēs parādīsim, kā instalēt TensorFlow ar NVIDIA CUDA/cuDNN paātrinājumu uz Debian 12 “Bookworm”.
Satura tēma:
- Pārbaude, vai datorā ir instalēts NVIDIA GPU
- Python 3 PIP un Python Venv instalēšana operētājsistēmā Debian 12
- Python 3 virtuālās vides izveide pakalpojumam TensorFlow
- Python 3 PIP jaunināšana Python 3 virtuālajā vidē
- TensorFlow instalēšana ar NVIDIA CUDA paātrinājuma atbalstu
- TensorRT instalēšana Debian 12
- TensorFlow Python 3 virtuālās vides aktivizēšana
- Piekļuve TensorFlow un pārbaude, vai ir pieejams NVIDIA GPU/CUDA paātrinājums
- Secinājums
Pārbaude, vai datorā ir instalēts NVIDIA GPU
Lai TensorFlow paātrinātu AI programmas ar NVIDIA GPU/CUDA, jums ir jābūt NVIDIA GPU draiveri un NVIDIA CUDA un cuDNN instalēta jūsu operētājsistēmā Debian 12.
Ja jums nepieciešama palīdzība NVIDIA GPU draiveru instalēšanā operētājsistēmā Debian 12, izlasi šo rakstu .
Ja jums nepieciešama palīdzība, instalējot NVIDIA CUDA un cuDNN draiverus savā Debian 12 operētājsistēmā, izlasi šo rakstu .
Kad jūsu Debian 12 sistēmā ir instalēti NVIDIA GPU draiveri, komandai “nvidia-smi” jābūt pieejamai.
NVIDIA kodola moduļi ir jāielādē arī jūsu Debian 12 sistēmā.
Kad esat instalējis NVIDIA CUDA draiverus, jūsu Debian 12 sistēmā jābūt pieejamai komandai “nvcc”.
Python 3 PIP un Python Venv instalēšana operētājsistēmā Debian 12
Lai instalētu TensorFlow uz Debian 12, ir jābūt instalētam Python 3 PIP un Python virtuālās vides (venv) modulim.
Vispirms atjauniniet APT pakotnes repozitorija kešatmiņu ar šādu komandu:
$ sudo piemērots atjauninājums 
Lai instalētu Python 3 PIP un Python 3 virtuālo vidi (venv), palaidiet šo komandu:
$ sudo apt uzstādīt python3-pip python3-venv python3-dev Lai apstiprinātu instalēšanu, nospiediet “Y” un pēc tam nospiediet
Tiek instalēts Python 3 PIP un Python 3 venv. Tas aizņem kādu laiku, lai pabeigtu.
Šajā brīdī ir jāinstalē Python 3 PIP un Python 3 venv.
Python 3 virtuālās vides izveide pakalpojumam TensorFlow
Standarta prakse Python bibliotēku instalēšanai Debian 12 ir to instalēšana Python virtuālajā vidē, lai tās netraucētu sistēmas Python pakotnēm/bibliotēkām.
Lai izveidotu jaunu Python 3 virtuālo vidi TensorFlow direktorijā “/opt/tensorflow”, palaidiet šo komandu:
$ sudo python3 -m venv / izvēlēties / tenzorplūsmaPython 3 PIP jaunināšana Python 3 virtuālajā vidē
Lai atjauninātu Python 3 PIP uz jaunāko versiju Python 3 virtuālajā vidē “/opt/tensorflow”, palaidiet šo komandu:
$ sudo / izvēlēties / tenzorplūsma / atkritumu tvertne / pip uzstādīt -- jauninājums pip 
TensorFlow instalēšana ar NVIDIA CUDA paātrinājuma atbalstu
Lai instalētu TensorFlow ar NVIDIA CUDA paātrinājuma atbalstu Python “/opt/tensorflow” virtuālajā vidē, palaidiet šādu komandu:
$ sudo / izvēlēties / tenzorplūsma / atkritumu tvertne / pip uzstādīt tenzorplūsma [ un-cuda ]Tiek instalēts TensorFlow ar NVIDIA CUDA paātrinājumu. Tas aizņem kādu laiku, lai pabeigtu.
Šajā brīdī ir jāinstalē TensorFlow ar NVIDIA CUDA paātrinājuma atbalstu.
TensorRT instalēšana Debian 12
NVIDIA TensorRT vēl vairāk optimizē TensorFlow dziļās mācīšanās veiktspēju. TensorRT var instalēt TensorFlow Python “/opt/tensorflow” virtuālajā vidē ar šādu komandu:
$ sudo / izvēlēties / tenzorplūsma / atkritumu tvertne / pip uzstādīt tensorrtNVIDIA TensorRT tiek instalēta Python virtuālajā vidē. Tas aizņem kādu laiku, lai pabeigtu.
Šajā brīdī ir jāinstalē NVIDIA TensorRT.
TensorFlow Python 3 virtuālās vides aktivizēšana
Lai aktivizētu TensorFlow Python “/opt/tensorflow” virtuālo vidi, palaidiet šādu komandu:
$ . / izvēlēties / tenzorplūsma / atkritumu tvertne / aktivizētIr jāaktivizē TensorFlow Python 3 virtuālā vide.
Piekļuve TensorFlow un pārbaude, vai ir pieejams NVIDIA GPU/CUDA paātrinājums
Lai atvērtu Python 3 interaktīvo čaulu, palaidiet šādu komandu:
$ python3Ir jāatver Python 3 interaktīvais apvalks.
Vispirms importējiet TensorFlow ar šādu koda rindiņu:
$ importa tensorflow kā tfKad TensorFlow ir importēts, varat pārbaudīt instalētās TensorFlow versijas numuru, izmantojot šo koda rindiņu. Kā redzat, mūsu Debian 12 sistēmā ir instalēta TensorFlow 2.13.1.
$ tf.__version__Lai pārbaudītu, vai TensorFlow var izmantot NVIDIA GPU, kuru instalējāt datorā CUDA paātrināšanai, izpildiet šo koda rindiņu. Kā redzat, mūsu NVIDIA GPU ir pieejams no TensorFlow.
$ drukāt ( tf.config.list_physical_devices ( 'GPU' ) ) 
Lai izietu no Python interaktīvās čaulas, palaidiet šo koda rindiņu:
$ atmest ( )Secinājums
Šajā rakstā mēs parādījām, kā instalēt Python 3 PIP un Python 3 virtuālo vidi (venv) Debian 12. Mēs arī parādījām, kā izveidot Python 3 virtuālo vidi TensorFlow operētājsistēmā Debian 12 un kā instalēt TensorFlow ar NVIDIA. GPU/CUDA paātrinājuma atbalsts un NVIDIA TensorRT arī operētājsistēmā Debian 12. Visbeidzot, mēs parādījām, kā aktivizēt TensorFlow Python virtuālo vidi un piekļūt TensorFlow operētājsistēmā Debian 12.