18 lines
338 B
YAML
18 lines
338 B
YAML
name: pytorch_sac_ae
|
|
channels:
|
|
- defaults
|
|
dependencies:
|
|
- python=3.7
|
|
- pytorch
|
|
- torchvision
|
|
- cudatoolkit=9.2
|
|
- absl-py
|
|
- pyparsing < 3.0.0
|
|
- pip:
|
|
- termcolor
|
|
- git+https://github.com/deepmind/dm_control.git
|
|
- git+https://github.com/denisyarats/dmc2gym.git
|
|
- tb-nightly
|
|
- imageio
|
|
- imageio-ffmpeg
|