Yaron Koresh commited on
Commit
609bd38
·
verified ·
1 Parent(s): 2b6b320

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -2
Dockerfile CHANGED
@@ -1,8 +1,10 @@
 
 
1
  WORKDIR /code
2
- COPY ./environment.yml /code/environment.yml
3
 
4
  # Create the environment using the environment.yml file
5
- RUN conda env create -f /code/environment.yml
6
 
7
  # Set up a new user named "user" with user ID 1000
8
  RUN useradd -m -u 1000 user
 
1
+ FROM nvidia/cuda
2
+
3
  WORKDIR /code
4
+ COPY ./env.yml /code/env.yml
5
 
6
  # Create the environment using the environment.yml file
7
+ RUN conda env create -f /code/env.yml
8
 
9
  # Set up a new user named "user" with user ID 1000
10
  RUN useradd -m -u 1000 user