1. 创建虚拟环境

conda create -n RL python=3.7

2. 激活虚拟环境

conda activate RL

3. 安装pytorch

# CPU
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cpuonly -c pytorch
# GPU
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge

4. 安装gym

pip install gym[atari]==0.25.2 -i https://mirrors.aliyun.com/pypi/simple

5. 安装学习需要的其他包

pip install -r requirements.txt

# pyyaml==6.0
# ipykernel==6.15.1
# jupyter==1.0.0
# matplotlib==3.5.3
# seaborn==0.12.1
# dill==0.3.5.1
# argparse==1.4.0
# pandas==1.3.5
# pyglet==1.5.26
# importlib-metadata<5.0
# setuptools==65.2.0

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐