dockerfile back to default

FLAIROx · Mar 28, 2024 · 4a37298 · 4a37298
1 parent 3da0866
commit 4a37298
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -18,7 +18,7 @@ RUN export XLA_PYTHON_CLIENT_MEM_FRACTION=0.25
 RUN export TF_FORCE_GPU_ALLOW_GROWTH=true
 
 # if you want jupyter 
-RUN pip install pip install jupyterlab
+# RUN pip install jupyterlab
 
 #for secrets and debug
 ENV WANDB_API_KEY=""

diff --git a/Makefile b/Makefile
@@ -1,10 +1,10 @@
-# NVCC_RESULT := $(shell which nvcc 2> NULL; rm NULL)
-# NVCC_TEST := $(notdir $(NVCC_RESULT))
-# ifeq ($(NVCC_TEST),nvcc)
+NVCC_RESULT := $(shell which nvcc 2> NULL; rm NULL)
+NVCC_TEST := $(notdir $(NVCC_RESULT))
+ifeq ($(NVCC_TEST),nvcc)
 GPUS=--gpus all
-# else
-# GPUS=
-# endif
+else
+GPUS=
+endif
 
 
 # Set flag for docker run command