diff --git a/AI/pytorch/2.0.1/23/Dockerfile b/AI/pytorch/2.0.1/23/Dockerfile index 1a5d1c7e865f53f05a6f3fd0410705d4176dad42..80f949fad34da63e5336ef5c7874bb7e3e36c8ba 100644 --- a/AI/pytorch/2.0.1/23/Dockerfile +++ b/AI/pytorch/2.0.1/23/Dockerfile @@ -3,8 +3,9 @@ FROM registry.openanolis.cn/openanolis/cuda:12.1.1-23-runtime LABEL \ maintainer="OpenAnolis AI SIG" \ org.opencontainers.image.title="pytorch" \ - org.opencontainers.image.version="2.0.1-23" + org.opencontainers.image.version="2.0.1-23" \ + org.opencontainers.image.date="20231115" -RUN dnf install pytorch-2.0.1 -y +RUN dnf install pytorch torch-vision -y CMD ["/bin/bash"] diff --git a/AI/tensorflow/2.12.0/23/Dockerfile b/AI/tensorflow/2.12.0/23/Dockerfile index e52b4d5de591b1a189de9c0dcc2d47cbf3745856..dafcd345a4c749441c85c2dcb290f2e4d0502b71 100644 --- a/AI/tensorflow/2.12.0/23/Dockerfile +++ b/AI/tensorflow/2.12.0/23/Dockerfile @@ -3,7 +3,8 @@ FROM registry.openanolis.cn/openanolis/cuda:12.1.1-23-runtime LABEL \ maintainer="OpenAnolis AI SIG" \ org.opencontainers.image.title="tensorflow" \ - org.opencontainers.image.version="2.12.0-23" + org.opencontainers.image.version="2.12.0-23" \ + org.opencontainers.image.date="20231115" RUN dnf install cuda-compiler-12-1 cuda-command-line-tools-12-1 cuda-cudart-devel-12-1 libcudnn -y RUN dnf install tensorflow2 -y