ARG OS_VERSION=20.04
ARG CUDA_VERSION=12.1.0
FROM --platform=${TARGETPLATFORM:-linux/amd64} nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${OS_VERSION}
ENV DEBIAN_FRONTEND=noninteractive
ENV RL_TOOLS_BACKEND_FLAGS="-DRL_TOOLS_BACKEND_ENABLE_CUDA=ON"
ENV CUDA_PATH=/usr/local/cuda-12.3
