J5Tsai commited on
Commit
1b48239
1 Parent(s): d3795c0

dev_fix_nchw_pass_cu121

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -8,7 +8,7 @@ EXPOSE 7860
8
 
9
  RUN apt-get update -y && apt-get install -y build-essential git curl wget tar python3 \
10
  && apt-get clean && rm -f /var/lib/apt/lists/*_*
11
- RUN python3 -m pip install -U --pre oneflow -f https://github.com/siliconflow/oneflow_releases/releases/expanded_assets/dev_fix_nchw_pass_cu118
12
  # The Space container runs with user ID 1000
13
  RUN useradd -m -u 1000 user
14
  ENV HOME=/home/user
 
8
 
9
  RUN apt-get update -y && apt-get install -y build-essential git curl wget tar python3 \
10
  && apt-get clean && rm -f /var/lib/apt/lists/*_*
11
+ RUN python3 -m pip install -U --pre oneflow -f https://github.com/siliconflow/oneflow_releases/releases/expanded_assets/dev_fix_nchw_pass_cu121
12
  # The Space container runs with user ID 1000
13
  RUN useradd -m -u 1000 user
14
  ENV HOME=/home/user