From 9a2446778045f27074c1054117b07e93f27cbd24 Mon Sep 17 00:00:00 2001
From: Myang-xhl <3038182679@qq.com>
Date: Fri, 7 Mar 2025 13:54:32 +0800
Subject: [PATCH] Update Dockerfile

---
 Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index a0febfa..b6636d1 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,10 +1,10 @@
-FROM pytorch/pytorch:2.3.1-cuda11.8-cudnn8-devel
+FROM docker pull pytorch/pytorch:2.6.0-cuda11.8-cudnn9-devel
 
 RUN apt-get update && apt-get install -y libgl1-mesa-glx libpci-dev curl nano psmisc zip git && apt-get --fix-broken install -y
 
 RUN conda install -y scikit-learn pandas flake8 yapf isort yacs future libgcc
 
 RUN pip install --upgrade pip && python -m pip install --upgrade setuptools && \
-    pip install opencv-python tb-nightly matplotlib logger_tt tabulate tqdm wheel mccabe scipy
+    pip install opencv-python tb-nightly matplotlib logger_tt tabulate tqdm wheel mccabe scipy numpy os mne
 
 COPY ./fonts/* /opt/conda/lib/python3.10/site-packages/matplotlib/mpl-data/fonts/ttf/