[Python知识库][RL] 使用 dockerfile 构建 atari 环境

强化学习需要 atari 环境，使用 dockerfile 快速构建环境。cuda11.2.2+ubuntu20.04+python3.8。

// 下载 ROMS
wget http://www.atarimania.com/roms/Roms.rar
unrar -e Roms.rar
unzip ROMS.zip
// 创建文件夹 atari
mkdir atari && cd atari
// 把 ROMS/ 复制过来
cp -r ../ROMS/ .
// 编辑 dockerfile
vim dockerfile

复制以下内容

FROM nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu20.04

MAINTAINER Jiale Han

SHELL ["/bin/bash","-c"]

WORKDIR /root

# ADD requirements.txt .
ADD ROMS/ ROMS/

RUN apt update \
    && apt install -y python3 python3-pip tmux \
    && pip install opencv-python matplotlib tqdm gym atari-py ale_py \
    && pip install torch==1.8.2+cu111 torchvision==0.9.2+cu111 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html \
    && python3 -m atari_py.import_roms ROMS/ \
    && apt install -y libgl1-mesa-glx \
    && DEBIAN_FRONTEND=noninteractive apt install -y libglib2.0-dev \
    && apt clean

构建镜像

cd atari
docker build -t nvidia/cuda:atari .

从镜像运行容器

docker run -it --gpus '"device=4"' -v /data9/hanjl:/root nvidia/cuda:atari /bin/bash