try
ci/woodpecker/push/woodpecker Pipeline failed Details

This commit is contained in:
Malte Grosse 2023-11-30 09:45:04 +09:00
parent 9ffb755608
commit c1c2795e42
2 changed files with 31 additions and 2 deletions

View File

@ -1,8 +1,8 @@
steps: steps:
"caffe": "caffe":
image: lablup/ngc-tensorflow:21.10-tf2-py3 image: nvcr.io/nvidia/tensorflow:23.10-tf2-py3
commands: commands:
- echo "ci working................b. " - echo "ci working................b. "
- sleep 100000 - python run.py

29
run.py Normal file
View File

@ -0,0 +1,29 @@
import tensorflow as tf
from tensorflow import keras
import matplotlib.pyplot as plt
import numpy as np
import os
# Version Information
# tensorflow 2.2.0 , Cudnn7.6.5 and Cuda 10.1 , python 3.8
from keras import backend as K
K.clear_session()
gpus = tf.config.experimental.list_physical_devices('GPU')
if gpus:
try:
tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=6024)])
except RuntimeError as e:
print(e)
os.exit(1)
print(tf.config.experimental.list_physical_devices())
print(tf.__version__)
print(tf.test.is_built_with_cuda())
(X_train, y_train), (X_test,y_test) = tf.keras.datasets.cifar10.load_data()
(X_train, y_train), (X_test,y_test) = tf.keras.datasets.cifar10.load_data()
print(X_train.shape,y_train.shape)