Fateme73 commited on
Commit
59e2fd5
·
verified ·
1 Parent(s): 82d5847

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -0
Dockerfile CHANGED
@@ -77,6 +77,11 @@ RUN mkdir /data && chown user:user /data
77
  # End root user section
78
  #######################################
79
 
 
 
 
 
 
80
 
81
  # Python packages
82
  RUN --mount=target=requirements.txt,source=requirements.txt \
 
77
  # End root user section
78
  #######################################
79
 
80
+ USER user
81
+
82
+ # Install TensorFlow 1.15 in Conda
83
+ RUN pip install --no-cache-dir tensorflow-gpu==1.15
84
+
85
 
86
  # Python packages
87
  RUN --mount=target=requirements.txt,source=requirements.txt \