import numpy as np
import pandas as pd
import torch

if __name__ == '__main__':
    print('Hello world')
    k = np.array([1,2,3])
    print(k)
    data = pd.DataFrame({'col1': [1,2,3], 'col2': [1,2,3]})
    print(data.head())
    print(torch.cuda.is_available())
