diff --git a/environment.yml b/environment.yml index 9817150..244fd84 100644 --- a/environment.yml +++ b/environment.yml @@ -14,6 +14,7 @@ dependencies: - pytorch::pytorch >=1.4.0 - cudatoolkit - cudnn + - ffmpeg - pytorch::cuda100 - matplotlib - git