From e73df5e50dd34a9fcea7aba7a786205a9386e099 Mon Sep 17 00:00:00 2001 From: InnopolisU Date: Tue, 27 Aug 2024 10:48:13 +0300 Subject: [PATCH] Updated dockerfiles in order for docker to be used with gpu --- DockerfileMED | 2 +- DockerfileNM | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/DockerfileMED b/DockerfileMED index cc8d5e5a..4755a376 100755 --- a/DockerfileMED +++ b/DockerfileMED @@ -1,4 +1,4 @@ -FROM python:3.8.3-slim +FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime ENV UI_TITLE="Фреймворк для Медицины" ENV WANDB_DISABLED="true" diff --git a/DockerfileNM b/DockerfileNM index 2b74a83d..33cb400c 100755 --- a/DockerfileNM +++ b/DockerfileNM @@ -1,4 +1,4 @@ -FROM python:3.8.3-slim +FROM pytorch/pytorch:1.11.0-cuda11.3-cudnn8-runtime ENV UI_TITLE="Фреймворк для Новых Материалов" ENV WANDB_DISABLED="true"