Skip to content

[devops] update pytorch version of CI #3724

@ver217

Description

@ver217

Overview

Our example tests and documentation tests use torch 1.12.0, see

image: hpcaitech/pytorch-cuda:1.12.0-11.3.0
.

However, out unit tests use torch 1.11.0, see

image: hpcaitech/pytorch-cuda:1.11.0-11.3.0
.

This inconsistency may lead to compatibility issue.

Goal

  1. Update torch verion of CI to 1.12.0
  2. Fix failed tests.

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions