File tree Expand file tree Collapse file tree 5 files changed +9
-6
lines changed
examples/v1beta1/trial-images
sdk/python/v1beta1/kubeflow/katib/api Expand file tree Collapse file tree 5 files changed +9
-6
lines changed Original file line number Diff line number Diff line change @@ -4,10 +4,12 @@ ENV TARGET_DIR /opt/darts-cnn-cifar10
4
4
5
5
ADD examples/v1beta1/trial-images/darts-cnn-cifar10 ${TARGET_DIR}
6
6
7
- WORKDIR ${TARGET_DIR}
7
+ WORKDIR ${TARGET_DIR}
8
8
9
+ # TODO (andreyvelich): This is required since torchvision==0.17.1 is incompatible with numpy 2.0
10
+ RUN pip install numpy==1.26.0
9
11
RUN pip install --prefer-binary --no-cache-dir torch==2.2.1 torchvision==0.17.1
10
- RUN pip install --prefer-binary --no-cache-dir -r requirements.txt
12
+
11
13
RUN chgrp -R 0 ${TARGET_DIR} \
12
14
&& chmod -R g+rwX ${TARGET_DIR}
13
15
Load Diff This file was deleted.
Original file line number Diff line number Diff line change @@ -6,6 +6,9 @@ WORKDIR /opt/pytorch-mnist
6
6
7
7
# Add folder for the logs.
8
8
RUN mkdir /katib
9
+
10
+ # TODO (andreyvelich): This is required since torchvision==0.17.1 is incompatible with numpy 2.0
11
+ RUN pip install numpy==1.26.0
9
12
RUN pip install --prefer-binary --no-cache-dir torch==2.2.1 torchvision==0.17.1
10
13
RUN pip install --prefer-binary --no-cache-dir -r requirements.txt
11
14
Original file line number Diff line number Diff line change 1
1
cloudml-hypertune == 0.1.0.dev6
2
- Pillow >= 9.1.1
Original file line number Diff line number Diff line change @@ -386,8 +386,8 @@ def tune(
386
386
image = base_image ,
387
387
command = ["bash" , "-c" ],
388
388
args = [exec_script ],
389
- env = env ,
390
- env_from = env_from ,
389
+ env = env if env else None ,
390
+ env_from = env_from if env_from else None ,
391
391
resources = resources_per_trial ,
392
392
)
393
393
],
You can’t perform that action at this time.
0 commit comments