dev_fix_nchw_pass_cu121

Dockerfile

@@ -8,7 +8,7 @@ EXPOSE 7860
 
 RUN apt-get update -y && apt-get install -y build-essential git curl wget tar python3 \
  && apt-get clean && rm -f /var/lib/apt/lists/*_*
-RUN python3 -m pip install -U --pre oneflow -f https://github.com/siliconflow/oneflow_releases/releases/expanded_assets/dev_fix_nchw_pass_cu118
+RUN python3 -m pip install -U --pre oneflow -f https://github.com/siliconflow/oneflow_releases/releases/expanded_assets/dev_fix_nchw_pass_cu121
 # The Space container runs with user ID 1000
 RUN useradd -m -u 1000 user
 ENV HOME=/home/user