diff --git a/README.md b/README.md index 506f0b3f..bfa1835a 100644 --- a/README.md +++ b/README.md @@ -20,6 +20,8 @@ For business inquiries, please visit our website and submit the form: [NVIDIA Re - `cd eg3d` - `conda env create -f environment.yml` - `conda activate eg3d` +* You can use a Docker environment that fits docker environments: + - `docker pull sunggukcha/eg3d` ## Getting started diff --git a/eg3d/environment.yml b/eg3d/environment.yml index 082bcaf5..e6441a5b 100644 --- a/eg3d/environment.yml +++ b/eg3d/environment.yml @@ -20,7 +20,7 @@ dependencies: - pillow=8.3.1 - scipy=1.7.1 - pytorch=1.11.0 - - cudatoolkit=11.1 + - cudatoolkit=11.3 - requests=2.26.0 - tqdm=4.62.2 - ninja=1.10.2 @@ -34,4 +34,4 @@ dependencies: - pyspng - psutil - mrcfile - - tensorboard \ No newline at end of file + - tensorboard