tkarras commited on
Commit
7bd6159
1 Parent(s): a6f45eb

Add notice about RTX 3090

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -63,7 +63,7 @@ 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.
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.
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"`.