additonal change (cf. google-research#224)

syncsyncsync · Nov 18, 2022 · 9eabb3c · 9eabb3c
1 parent 6fc8ed3
commit 9eabb3c
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/docker/Kubruntu.Dockerfile b/docker/Kubruntu.Dockerfile
@@ -6,9 +6,19 @@
 #   --workdir "/kubric" \
 #   kubricdockerhub/kubruntu:latest \
 #   python3 examples/helloworld.py
+ARG HTTP_PROXY
+ARG HTTPS_PROXY
+ARG http-proxy-host
+ARG http-proxy-port 
+ARG NO_PROXY="10/8,172.16/12,192.168/16"
 
-FROM kubricdockerhub/blender:latest
+FROM kubricdockerhub/blender_gpu:latest
 
+ENV HTTP_PROXY=$HTTPS_PROXY
+ENV http_proxy=$HTTP_PROXY
+ENV HTTPS_PROXY=$HTTPS_PROXY
+ENV https_proxy=$HTTPS_PROXY
+ENV NO_PROXY=$NO_PROXY
 WORKDIR /kubric
 
 # --- copy requirements in workdir

diff --git a/docker/enable_cuda_patch.txt b/docker/enable_cuda_patch.txt
@@ -0,0 +1,8 @@
+--- a/build_files/cmake/config/bpy_module.cmake
++++ b/build_files/cmake/config/bpy_module.cmake
+@@ -44,3 +44,5 @@
+ if(WIN32)
+   set(WITH_WINDOWS_BUNDLE_CRT  OFF CACHE BOOL "" FORCE)
+ endif()
++
++set(WITH_CYCLES_CUDA_BINARIES   ON  CACHE BOOL "" FORCE)