1. 创建conda新环境
conda create --name gymTorch python=3.7
conda activate gymTorch(进入新环境)
(python如果为3.6版本,在导入gym包时会报错 AttributeError: module 'contextlib' has no attribute 'nullcontext')
2. 安装pytorch
进入官网,选择合适的pytorch版本。官网会根据用户的选择提供相应安装torch的命令,将该命令复制到conda命令行中执行即可。
3. 安装gym
官网建议用pip install gym的方式安装。但是在conda环境中用该命令一般是会安装到base环境,所以需要将gym相关文件转移到conda环境的site-packages中。
3. 测试包是否成功导入
在conda的gymTorch环境中,依次输入以下命令,没有报错就是导入正常
python
import torch
import gym