Skip to content

sample-pytorchjob should update to cuda12+ and pytorch 2.0 #1910

@village-way

Description

@village-way

What would you like to be added:

sample-pytorchjob should update cuda and pytorch version, the image in the examples/jobs/sample-pytorchjob.yaml is too old.

Why is this needed:

the sample pytorchjob failed to run on ubuntu 22.04 with cuda12.3

Completion requirements:

This enhancement requires the following artifacts:
the image docker.io/kubeflowkatib/pytorch-mnist:v1beta1-45c5727 is from kubeflowkatib and the dockerfile is updated by kubeflow/katib#2278

The artifacts should be linked in subsequent comments.

Metadata

Metadata

Assignees

No one assigned

    Labels

    kind/featureCategorizes issue or PR as related to a new feature.

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions