-
copy .env.example to .env
$ cp .env.example .env
- If you want to enable GPUs, set the .env file as follows
TARGET_ARCH=gpu RUNTIME=nvidia TARGET_IMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
- If you use the CPU, set the .env file as follows
TARGET_ARCH=cpu
- If you want to enable GPUs, set the .env file as follows
-
build docker image
$ docker-compose build
-
run docker container
$ docker-compose up -d
-
exec learning
$ docker-compose exec rl_adjust_pose bash $ python3 src/learning.py
-
exec tensorboard
$ docker exec rl_adjust_pose tensorboard --logdir logs
- If X forwading is failing, try the following command on the host side.
$ xhost +local: