Martí Umbert
commited on
Commit
·
3ed927c
1
Parent(s):
4c3038e
Dockerfile: change base image, keep installation of libaio1 libaio-dev
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
# ORI --> cuDNN error: CUDNN_STATUS_SUBLIBRARY_VERSION_MISMATCH
|
| 2 |
-
|
| 3 |
# TESTS UNTIL RIGHT PYTHON VERSION
|
| 4 |
# FROM nvidia/cuda:11.6.1-devel-ubuntu20.04
|
| 5 |
# FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu18.04
|
|
@@ -7,7 +7,7 @@
|
|
| 7 |
# FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
|
| 8 |
# FROM nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu22.04
|
| 9 |
# FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
|
| 10 |
-
FROM nvcr.io/nvidia/pytorch:24.07-py3
|
| 11 |
|
| 12 |
ENV DEBIAN_FRONTEND=noninteractive
|
| 13 |
ENV PYTHONUNBUFFERED=1
|
|
|
|
| 1 |
# ORI --> cuDNN error: CUDNN_STATUS_SUBLIBRARY_VERSION_MISMATCH
|
| 2 |
+
FROM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu22.04
|
| 3 |
# TESTS UNTIL RIGHT PYTHON VERSION
|
| 4 |
# FROM nvidia/cuda:11.6.1-devel-ubuntu20.04
|
| 5 |
# FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu18.04
|
|
|
|
| 7 |
# FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04
|
| 8 |
# FROM nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu22.04
|
| 9 |
# FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
|
| 10 |
+
# FROM nvcr.io/nvidia/pytorch:24.07-py3
|
| 11 |
|
| 12 |
ENV DEBIAN_FRONTEND=noninteractive
|
| 13 |
ENV PYTHONUNBUFFERED=1
|