Commit
•
9cee656
1
Parent(s):
39b6d10
changed docker image
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -13,7 +13,7 @@ RUN pip install --no-cache-dir torch torchvision
|
|
| 13 |
|
| 14 |
# Add and run the GPU check script
|
| 15 |
COPY gpu_check.py /code/gpu_check.py
|
| 16 |
-
RUN
|
| 17 |
|
| 18 |
# Set up a new user named "user" with user ID 1000
|
| 19 |
RUN useradd -m -u 1000 user
|
|
|
|
| 13 |
|
| 14 |
# Add and run the GPU check script
|
| 15 |
COPY gpu_check.py /code/gpu_check.py
|
| 16 |
+
RUN python3 /code/gpu_check.py
|
| 17 |
|
| 18 |
# Set up a new user named "user" with user ID 1000
|
| 19 |
RUN useradd -m -u 1000 user
|