diff --git a/Dockerfile b/Dockerfile index 339f18f..3e2296e 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,5 @@ -FROM conda/miniconda3 +FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime +#FROM tensorflow/tensorflow:2.12.0-gpu ARG USER ARG GROUP diff --git a/daemon.sh b/daemon.sh old mode 100644 new mode 100755 diff --git a/interactively.sh b/interactively.sh old mode 100644 new mode 100755