From cac382ef19e1345c5742b73384b531a6303307ed Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Tue, 6 Jun 2023 22:17:49 +0200 Subject: [PATCH] docker update --- docker/Dockerfile | 2 +- docker/docker-compose.yml | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index a2cdf54..96f3d55 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,6 +1,6 @@ # Use an existing image as the base # FROM nvcr.io/nvidia/pytorch:23.04-py3 -FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-devel +FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel # FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel # Add the current user to the image diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml index 240cd90..d34bde0 100644 --- a/docker/docker-compose.yml +++ b/docker/docker-compose.yml @@ -40,8 +40,8 @@ services: - capabilities: [gpu] lhunor_torch_3: - image: lhunor_reconstruction:v2.0 - container_name: lhunor_reconstruction_v2.0 + image: lhunor_reconstruction:v2.0.1 + container_name: lhunor_reconstruction_v2.0.1 ipc: host tty: true ulimits: