Skip to content

Commit 28b848d

Browse files
committed
add pip install instructions to README
1 parent dd06bf3 commit 28b848d

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

README.md

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,24 @@
1+
## StyleGAN2-ADA — `pip install` version of Official PyTorch implementation
2+
3+
I have modified the official PyTorch implementation so that you can `pip install` this repository as a dependency and reuse the classes and functions here.
4+
5+
### Requirements
6+
7+
* Linux and Windows are supported, but we recommend Linux for performance and compatibility reasons.
8+
* 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.
9+
* 64-bit Python 3.7 and PyTorch 1.7.1. See [https://pytorch.org/](https://pytorch.org/) for PyTorch install instructions.
10+
* 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).)
11+
12+
### Installation
13+
14+
From repo's root directory `stylegan2-ada-pytorch`, run `python -m pip install .`
15+
16+
### Original official implementation
17+
18+
Available [here](https://github.com/NVlabs/stylegan2-ada-pytorch), the original `README.md` is copied below.
19+
20+
***
21+
122
## StyleGAN2-ADA — Official PyTorch implementation
223

324
![Teaser image](./docs/stylegan2-ada-teaser-1024x252.png)

0 commit comments

Comments
 (0)