Эта инструкция посвящена установке и настройке библиотеке gymnasium, а также основам работы с этой библиотекой.

Библиотека gymnasium предназначена для разработки и сравнения алгоритмов обучения с подкреплением (Reinforcement Learning, RL) и предоставляет множество готовых окружений (Environments), в которых будут действовать агенты.

mkdir ~/Projects/gym-tf-tmp
cd ~/Projects/gym-tf-tmp
python -m venv venv
source ./venv/bin/activate

Установить gymnasium сам по себе: pip install gymnasium. Будут установлены почти все зависимости Classic Control-окружений. Это самые базовые окружения. Можно будет создавать эти окружения:

import gymnasium as gym
env = gym.make('CartPole-v1')

Если запустить установку, специфическую для Classic Control-окружений, то будет доставлена библиотека pygame:

pip install gymnasium[classic-control]

Чтобы иметь возможность создавать окружения из набора Box2D (это простые игры, основанные на 2D-физике, предоставляемой библиотекой box2d, и рендеринге, основанном на pygame), нужно сначала установить библиотеку swig, предназначенную для соединения программ, написанных на C и C++, с другими языками программирования, в т.ч. Python:

sudo apt install swig  # need for box2d-py

После чего можно запустить установку gymnasium с Box2D-окружениями:

pip install gymnasium[box2d]

Чтобы иметь возможность сохранять видео (используя wrappers.RecordVideo), нужно доустановить пакет moviepy:

pip install moviepy

Можно установить все возможности gymnasium, но в этом случае он поставить библиотеки nvidia даже на компьютере без видеокарт Nvidia.

pip install gymnasium[all]