environment.yaml

name: pytorch1.7.1
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - cudatoolkit=11.0.3=h15472ef_9
  - matplotlib=3.2.2=1
  - matplotlib-base=3.2.2=py37h1d35a4c_1
  - mpi=1.0=mpich
  - mpi4py=3.1.3=py37h1e5cb63_0
  - numpy=1.20.3=py37hf144106_0
  - numpy-base=1.20.3=py37h74d4b33_0
  - python=3.7.12=hf930737_100_cpython
  - python-dateutil=2.8.2=pyhd8ed1ab_0
  - python_abi=3.7=2_cp37m
  - pytorch=1.7.1=py3.7_cuda11.0.221_cudnn8.0.5_0
  - pyyaml=6.0=py37h540881e_4
  - torchaudio=0.7.2=py37
  - torchvision=0.8.2=py37_cu110
  - wheel=0.37.1=pyhd8ed1ab_0
  - yaml=0.2.5=h7f98852_2
  - pip:
    - gym==0.15.7
    - mujoco==2.3.2
    - tensorboard==2.10.0
    - tensorboard-data-server==0.6.1
    - tensorboard-plugin-wit==1.8.1
    - tensorboardx==2.2
    - tqdm==4.62.3