15 lines
337 B
YAML
15 lines
337 B
YAML
name: drl_hw2
|
|
channels:
|
|
- pytorch
|
|
- nvidia
|
|
- defaults
|
|
dependencies:
|
|
- python=3.10
|
|
- pytorch=2.0.0
|
|
- pytorch-cuda=11.7 # Comment this line if you are a mac user or want a cpu-only installation
|
|
- pip=23.0.1
|
|
- pip:
|
|
- gymnasium[classic-control]==0.27.1
|
|
- hydra-core==1.3.2
|
|
- matplotlib==3.7.1
|
|
- moviepy==1.0.3 |