Skip to content

Commit

Permalink
Update nerfstudio Env
Browse files Browse the repository at this point in the history
  • Loading branch information
LingzheZhao committed May 27, 2023
1 parent 111e680 commit c47b981
Show file tree
Hide file tree
Showing 3 changed files with 23 additions and 22 deletions.
10 changes: 4 additions & 6 deletions Example_Envs/nerfstudio/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,10 +1,5 @@
FROM nvcr.io/nvidia/cuda:11.8.0-devel-ubuntu22.04 AS base

LABEL org.opencontainers.image.version = "0.5.21"
LABEL org.opencontainers.image.source = "https://github.com/WU-CVGL/cluster-user-guide"
LABEL org.opencontainers.image.licenses = "Apache License 2.0"
LABEL org.opencontainers.image.base.name="nvcr.io/nvidia/cuda:11.8.0-devel-ubuntu22.04"

ARG MAX_JOBS 32
ARG DEBIAN_FRONTEND=noninteractive
ARG CUDA_ARCHITECTURES=89;86
Expand Down Expand Up @@ -139,7 +134,10 @@ RUN pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 --extra-index-url h
rm -rf /tmp/*

RUN pip install open3d>=0.16.0 --ignore-installed &&\
pip install nerfstudio==0.3.0 -i https://pypi.org/simple/ &&\
# Stable version
# pip install nerfstudio==0.3.1 -i https://pypi.org/simple/ &&\
# Nightly version
pip install git+https://github.com/nerfstudio-project/nerfstudio@main -i https://pypi.org/simple/ &&\
pip install einops torchtyping>=0.1.4 kplanes-nerfstudio &&\
rm -rf /tmp/*

Expand Down
2 changes: 1 addition & 1 deletion Example_Envs/nerfstudio/Makefile
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
NERF_NAME=harbor.cvgl.lab/library/zlz-nerfstudio
NERF_VERSION=0.3.0-cuda-11.8-devel-ubuntu22.04-torch-2.0.1-230522
NERF_VERSION=nightly-cuda-11.8-devel-ubuntu22.04-torch-2.0.1-230527
NERF=$(NERF_NAME):$(NERF_VERSION)

SDF_NAME=harbor.cvgl.lab/library/zlz-sdfstudio
Expand Down
33 changes: 18 additions & 15 deletions Home.md
Original file line number Diff line number Diff line change
Expand Up @@ -72,10 +72,11 @@ GPU Node 1:
| :----: | :---- |
| Model | Powerleader PR4908R (Supermicro 4124GS-TNR)|
| CPU | AMD EPYC 7302 * 2 (32C/64T, 3.0-3.3GHz)|
| Memory | Samsung M393A2K43DB2-CVF DDR4 256G (16G*16) 2933MT/s ECC REG|
| GPU | MSI RTX 3090 Turbo * 8 |
| RAM | Samsung M393A2K43DB2-CVF DDR4 256G (16G*16) 2933MT/s ECC REG|
| GPU | MSI (0x1462) RTX 3090 Turbo * 8 |
| SSD | Intel P4510 2TB * 1 |
| NIC | Intel 82599ES 10GbE |
| NIC | Intel I350-T2 1GbE Dual Port|
| NIC | Intel X520-SR2 (82599ES) 10GbE Dual Port |
| RAID | LSI MegaRAID SAS-3 3108 |

GPU Node 2, 3, 4:
Expand All @@ -84,10 +85,11 @@ GPU Node 2, 3, 4:
| :----: | :---- |
| Model | Powerleader PR4908R (Supermicro 4124GS-TNR)|
| CPU | AMD EPYC 7402 * 2 (48C/96T, 2.8-3.35GHz)|
| Memory | SK Hynix† / Samsung‡ / Samsung‡ DDR4 512G (32G*16) 3200MT/s ECC REG|
| GPU | NVIDIA / MSI / MSI RTX 3090 * 8 |
| RAM | SK Hynix† / Samsung‡ / Samsung‡ DDR4 512G (32G*16) 3200MT/s ECC REG|
| GPU | NVIDIA (0x10DE) / MSI (0x1462) / MSI (0x1462) RTX 3090 * 8 |
| SSD | Intel P4510 2TB * 1 |
| NIC | Intel 82599ES 10GbE |
| NIC | Intel I350-T2 1GbE Dual Port|
| NIC | Intel X520-SR2 (82599ES) 10GbE Dual Port |

> † SK Hynix 3200: HMA84GR7DJR4N-XN
>
Expand All @@ -99,11 +101,11 @@ GPU Node 5:
| :----: | :---- |
| Model | ASUS ESC8000A-E11|
| CPU | AMD EPYC 7543 * 2 (64C/128T, 2.8-3.7GHz)|
| Memory | Samsung M393A4K40EB3-CWE DDR4 512G (32G*16) 3200MT/s ECC REG|
| GPU | MANLI RTX 4090 * 8 |
| RAM | Samsung M393A4K40EB3-CWE DDR4 512G (32G*16) 3200MT/s ECC REG|
| GPU | MANLI (NVIDIA/0x10DE) RTX 4090 * 8 |
| SSD | Intel S4610 (SSDSC2KG96) 960G * 2 (RAID 1) |
| NIC | Intel I350-T4 |
| NIC | Intel 82599ES 10GbE Dual Port |
| NIC | Intel I350-T4 1GbE Quad Port|
| NIC | Intel X520-SR2 (82599ES) 10GbE Dual Port |
| RAID | LSI SAS3008 PCI-Express Fusion-MPT SAS-3 |

Storage Server
Expand All @@ -112,11 +114,12 @@ Storage Server
| :----: | :---- |
| Model | Powerleader PR4224AK (Supermicro H11SSL)|
| CPU | AMD EPYC 7302 (16C/32T, 3.0-3.3GHz)|
| Memory | Samsung M393A4K40DB3-CWE DDR4 256G (32G*8) 2933MT/s ECC REG |
| SSD | Samsung 970 EVO Plus 500G * 1|
| RAM | Samsung M393A4K40DB2-CWE DDR4 256G (32G*8) 2933MT/s ECC REG |
| SSD | INTEL 760p (SSDPEKKW256G8) 256G * 1|
| SSD | Intel S4510 1.92TB * 2 |
| HDD | Seagate Exos X18 18TB * 14 |
| NIC | Intel 82599ES 10GbE Dual Port |
| NIC | Intel i210 * 2 |
| NIC | Intel X520-SR2 (82599ES) 10GbE Dual Port |
| RAID | LSI SAS3008 PCI-Express Fusion-MPT SAS-3 |

Management Server
Expand All @@ -125,9 +128,9 @@ Management Server
| :----: | :---- |
| Model | ASUS RS520-E9-RS8 V2 |
| CPU | Intel Xeon Silver 4210R * 2 (20C/40T, 2.4-3.2GHz) |
| Memory | Samsung M393A4K40EB3-CWE DDR4 64G (32G*2) 3200MT/s @ 2400MT/s ECC REG |
| RAM | Samsung M393A4K40EB3-CWE DDR4 64G (32G*2) 3200MT/s @ 2400MT/s ECC REG |
| SSD | Intel S4610 (SSDSC2KG96) 960G * 2 (RAID 1) |
| NIC | Intel i350-AM2 Dual Port |
| NIC | Intel i350-AM2 1GbE Dual Port |
| RAID | LSI SAS3008 PCI-Express Fusion-MPT SAS-3 |

<details>
Expand Down

0 comments on commit c47b981

Please sign in to comment.