update
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
FROM nvidia/cuda:11.1.
|
| 2 |
|
| 3 |
ENV CUDA_HOME=/usr/local/cuda
|
| 4 |
ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:${PATH}
|
|
|
|
| 1 |
+
FROM nvidia/cuda:11.1.1-devel-ubuntu20.04
|
| 2 |
|
| 3 |
ENV CUDA_HOME=/usr/local/cuda
|
| 4 |
ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:${PATH}
|