diff --git a/README.md b/README.md index 092b52bb4..f034ed8e1 100755 --- a/README.md +++ b/README.md @@ -65,7 +65,7 @@ This repository is a faithful reimplementation of [StyleGAN2-ADA](https://github * 1–8 high-end NVIDIA GPUs with at least 12 GB of memory. We have done all testing and development using NVIDIA DGX-1 with 8 Tesla V100 GPUs. * 64-bit Python 3.7 and PyTorch 1.7.1. See [https://pytorch.org/](https://pytorch.org/) for PyTorch install instructions. * CUDA toolkit 11.0 or later. Use at least version 11.1 if running on RTX 3090. (Why is a separate CUDA toolkit installation required? See comments in [#2](https://github.com/NVlabs/stylegan2-ada-pytorch/issues/2#issuecomment-779457121).) -* Python libraries: `pip install click requests tqdm pyspng ninja imageio-ffmpeg==0.4.3`. We use the Anaconda3 2020.11 distribution which installs most of these by default. +* Python libraries: `pip install click requests tqdm pyspng ninja imageio imageio-ffmpeg==0.4.3`. We use the Anaconda3 2020.11 distribution which installs most of these by default. * Docker users: use the [provided Dockerfile](./Dockerfile) to build an image with the required library dependencies. The code relies heavily on custom PyTorch extensions that are compiled on the fly using NVCC. On Windows, the compilation requires Microsoft Visual Studio. We recommend installing [Visual Studio Community Edition](https://visualstudio.microsoft.com/vs/) and adding it into `PATH` using `"C:\Program Files (x86)\Microsoft Visual Studio\\Community\VC\Auxiliary\Build\vcvars64.bat"`.