15 lines
289 B
YAML
15 lines
289 B
YAML
|
name: pytorch_sac_ae
|
||
|
|
channels:
|
||
|
|
- defaults
|
||
|
|
dependencies:
|
||
|
|
- python=3.6
|
||
|
|
- pytorch
|
||
|
|
- torchvision
|
||
|
|
- cudatoolkit=9.2
|
||
|
|
- pip:
|
||
|
|
- colored
|
||
|
|
- absl-py
|
||
|
|
- git+git://github.com/deepmind/dm_control.git
|
||
|
|
- git+git://github.com/1nadequacy/dmc2gym.git
|
||
|
|
- tb-nightly
|
||
|
|
- imageio
|