Janne Hellsten commited on
Commit
bb409d8
1 Parent(s): 7a7a294

Update README.md python requirements (#2)

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -63,7 +63,8 @@ This repository is a faithful reimplementation of [StyleGAN2-ADA](https://github
63
 
64
  * Linux and Windows are supported, but we recommend Linux for performance and compatibility reasons.
65
  * 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.
66
- * 64-bit Python 3.7, PyTorch 1.7.1, and CUDA toolkit 11.0 or newer. Use CUDA toolkit 11.1 or later with RTX 3090.
 
67
  * Docker users: use the [provided Dockerfile](./Dockerfile) to build an image with the required library dependencies.
68
 
69
  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\<VERSION>\Community\VC\Auxiliary\Build\vcvars64.bat"`.
 
63
 
64
  * Linux and Windows are supported, but we recommend Linux for performance and compatibility reasons.
65
  * 1&ndash;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.
66
+ * 64-bit Python 3.7, PyTorch 1.7.1, and CUDA toolkit 11.0 or newer. Use CUDA toolkit 11.1 or later with RTX 3090. See [https://pytorch.org/](https://pytorch.org/) for PyTorch install instructions.
67
+ * 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.
68
  * Docker users: use the [provided Dockerfile](./Dockerfile) to build an image with the required library dependencies.
69
 
70
  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\<VERSION>\Community\VC\Auxiliary\Build\vcvars64.bat"`.