From 74888a4c29899763984b9de99fa238912ebf9117 Mon Sep 17 00:00:00 2001 From: Realcat Date: Sun, 22 Sep 2024 21:49:05 +0800 Subject: [PATCH] Dev: update dockerfile (#76) * update: env * update: env * update: dockerfile * update: numpy version --- Dockerfile | 2 +- requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 863cdf8..7455862 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,5 +1,5 @@ # Use an official conda-based Python image as a parent image -FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime +FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime LABEL maintainer vincentqyw ARG PYTHON_VERSION=3.10.10 diff --git a/requirements.txt b/requirements.txt index e979c2f..4ec3af7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -11,7 +11,7 @@ Jinja2 kornia loguru matplotlib -numpy +numpy==1.23.5 onnxruntime omegaconf opencv-python