From bf462eef453739bfe976d9de3330a120054514bf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Erik=20H=C3=A4rk=C3=B6nen?= Date: Tue, 23 Jun 2020 13:49:48 +0300 Subject: [PATCH] SG2: added custom OP fallbacks for CPU support --- SETUP.md | 5 +++-- models/stylegan2/stylegan2-pytorch | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/SETUP.md b/SETUP.md index 8aff461..23bce04 100644 --- a/SETUP.md +++ b/SETUP.md @@ -23,8 +23,9 @@ Install included dependencies (downloaded from https://www.lfd.uci.edu/~gohlke/p 5. Compile and install: `make install` 6. Install Glumpy: `pip install setuptools cython glumpy` -### StyleGAN2 -The bundled StyleGAN2 model requires additional setup steps due to the custom CUDA kernels involved. +### StyleGAN2 setup (optional) +StyleGAN2 contains custom CUDA kernels for improved performance.
+Less performant native PyTorch fallbacks are used by default. 1. Install CUDA toolkit (match the version in environment.yml) 2. On Windows: install and open 'x64 Native Tools Command Prompt for VS 2017' 3. `conda activate ganspace` diff --git a/models/stylegan2/stylegan2-pytorch b/models/stylegan2/stylegan2-pytorch index 5706513..76ca105 160000 --- a/models/stylegan2/stylegan2-pytorch +++ b/models/stylegan2/stylegan2-pytorch @@ -1 +1 @@ -Subproject commit 5706513b23e8240480395faefb38f3150ceccd3d +Subproject commit 76ca1058fc587b1846523bc382f0ece848bb0711