深度强化学习gym和pytorch安装最全教程
深度强化学习gym和pytorch安装最全教程
·
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

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