Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
# Use NVIDIA CUDA base image
|
| 2 |
-
|
| 3 |
|
| 4 |
# Set the working directory
|
| 5 |
WORKDIR /code
|
|
|
|
| 1 |
# Use NVIDIA CUDA base image
|
| 2 |
+
RUN --gpus all nvidia/cuda:10.2-cudnn7-devel nvidia-smi
|
| 3 |
|
| 4 |
# Set the working directory
|
| 5 |
WORKDIR /code
|