15 lines
405 B
YAML
15 lines
405 B
YAML
# Copyright Broadcom, Inc. All Rights Reserved.
|
|
# SPDX-License-Identifier: APACHE-2.0
|
|
|
|
command:
|
|
check-pytorch-version:
|
|
exec: python -c 'import torch; print(torch.__version__)'
|
|
timeout: 30000
|
|
exit-status: 0
|
|
stdout:
|
|
- "{{ .Env.APP_VERSION }}"
|
|
check-torchvision-version:
|
|
exec: python -c 'import torchvision; print(torchvision.__version__)'
|
|
timeout: 30000
|
|
exit-status: 0
|