Установка TensorFlow в Blender

TensorFlow – это библиотека с открытым исходным кодом для создания и обучения нейронных сетей от Google.

Установить TensorFlow в Blender можно при помощи установщика пакетов pip.

Перед началом установки убедитесь, что ваша система соответствует минимальным требованиям:

Blender должен быть версии 2.80 или старше (минимальная требуемая версия Python в Blender – 3.7)
Ваша операционная система должна быть:
- Windows 7 (x64) или старше
- Ubuntu 16.04 (x64) или старше
- MacOS 10.12.6 (x64) или старше

Для установки под Windows возможно потребуется дополнительно установить MS Visual C++ Redistributable for Visual Studio

Для установки TensorFlow в Blender воспользуемся скриптом для установки пакетов, указав в названии пакета “tensorflow”.

import subprocess
import sys
import os
 
python_exe = os.path.join(sys.prefix, 'bin', 'python.exe')
target = os.path.join(sys.prefix, 'lib', 'site-packages')

# upgrade pip
subprocess.call([python_exe, '-m', 'ensurepip'])
subprocess.call([python_exe, '-m', 'pip', 'install', '--upgrade', 'pip'])
 
# install tensorflow package
subprocess.call([python_exe, '-m', 'pip', 'install', '--upgrade', 'tensorflow', '-t', target])

print('TensorFlow installed')

import subprocess

import sys

import os

python_exe = os.path.join(sys.prefix, 'bin', 'python.exe')

target = os.path.join(sys.prefix, 'lib', 'site-packages')

# upgrade pip

subprocess.call([python_exe, '-m', 'ensurepip'])

subprocess.call([python_exe, '-m', 'pip', 'install', '--upgrade', 'pip'])

# install tensorflow package

subprocess.call([python_exe, '-m', 'pip', 'install', '--upgrade', 'tensorflow', '-t', target])

print('TensorFlow installed')

Откройте окно Text Editor, вставьте в него этот код и запустите его, нажав на кнопку со стрелочкой “Run Script”.

Проверить доступность TensorFlow в Blender после установки можно выполнив из Python-консоли или Text Editor следующий код:

import tensorflow as tf

print(tf.reduce_sum(tf.random.normal([1000, 1000])))

# tf.Tensor(-261.76862, shape=(), dtype=float32)

import tensorflow as tf

print(tf.reduce_sum(tf.random.normal([1000, 1000])))

# tf.Tensor(-261.76862, shape=(), dtype=float32)

Если ваша видеокарта поддерживает технологию CUDA, TensorFlow может проводить вычисления с использованием GPU. Для этого потребуется дополнительно установить:

Драйвера NVIDIA с поддержкой GPU
CUDA Toolkit 11.2 или старше
CUPTI
cuDNN SDK 8.1.0 или старше
(при необходимости) TensorRT 6.0 или старше

Подробности по установке этих приложений приведены на официальном сайте TensorFlow.

После установки перечисленных приложений, TensorFlow должна самостоятельно подключить GPU к работе.