Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
@@ -1,2 +1,3 @@
|
|
1 |
-
FROM
|
2 |
|
|
|
|
1 |
+
FROM pytorch/pytorch:1.8.1-cuda11.1-cudnn8-devel
|
2 |
|
3 |
+
RUN pip install SimpleITK
|