verify_setup.py

import torch
import transformers
import sacrebleu
import jieba
import matplotlib
import seaborn
import tqdm
import json
import os

print(f"PyTorch version: {torch.__version__}")
print(f"CUDA available: {torch.cuda.is_available()}")
if torch.cuda.is_available():
    print(f"CUDA device: {torch.cuda.get_device_name(0)}")

print("All imports successful!")

# Check data access
data_path = "/250010108/nlp/dataset/AP0004_Midterm&Final_translation_dataset_zh_en/train_10k.jsonl"
if os.path.exists(data_path):
    print(f"Data found at {data_path}")
    with open(data_path, 'r') as f:
        print(f"First line of data: {f.readline().strip()}")
else:
    print(f"Data NOT found at {data_path}")