Re-release 2.7.0 (#578)

ServiceNow · Jun 12, 2023 · fa7fc5e · fa7fc5e
2 parents a2621e7 + f55e7bc
commit fa7fc5e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -1,6 +1,6 @@
 ARG DEVICE=cpu
 
-FROM pytorch/pytorch:1.8.1-cuda11.1-cudnn8-runtime as build_gpu
+FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-runtime as build_gpu
 # Copy binaries from other images here
 RUN pip install --upgrade pip