From c1c86e1d6921ee5d5e8f834f15e2fcc992ad4251 Mon Sep 17 00:00:00 2001
From: siaimes <34199488+siaimes@users.noreply.github.com>
Date: Thu, 8 Jun 2023 22:10:38 +0800
Subject: [PATCH] 1.8.1-cuda11.1-cudnn8-devel

Signed-off-by: siaimes <34199488+siaimes@users.noreply.github.com>
---
 Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 9501657..a49cfdb 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.7.1-cuda11.0-cudnn8-devel
+FROM pytorch/pytorch:1.8.1-cuda11.1-cudnn8-devel
 
 RUN apt-key del 7fa2af80 && \
     apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub && \