Spaces:
Running
Running
unknown
commited on
Commit
•
7bd6159
1
Parent(s):
a6f45eb
Add notice about RTX 3090
Browse files
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–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"`.
|