radames commited on
Commit
c37fd8b
1 Parent(s): f3a65fd
Files changed (3) hide show
  1. Dockerfile +9 -0
  2. README.md +1 -0
  3. requirements.txt +4 -0
Dockerfile ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ FROM dromni/nerfstudio:0.3.2
2
+
3
+ ARG CUDA_VERSION=11.8.0
4
+ ARG CUDA_ARCHITECTURES=86
5
+ ARG OS_VERSION=22.04
6
+
7
+ RUN ["ns-download-data", "nerfstudio", "--capture-name=poster"]
8
+
9
+ CMD ["ns-train", "nerfacto", "--data", "data/nerfstudio/poster"]
README.md CHANGED
@@ -5,6 +5,7 @@ colorFrom: red
5
  colorTo: green
6
  sdk: docker
7
  pinned: false
 
8
  ---
9
 
10
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
5
  colorTo: green
6
  sdk: docker
7
  pinned: false
8
+ app_port: 7007
9
  ---
10
 
11
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ --extra-index-url https://download.pytorch.org/whl/cu118
2
+ torch==2.0.1+cu118
3
+ torchvision==0.15.2+cu118
4
+ git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch