test.py

import torch
print(torch.__version__)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)
print(torch.version.hip)
print(torch.cuda.get_device_name(0))
print(torch.cuda.device_count())
import vllm
print(vllm.__version__)
import multiprocessing

cpu_count = multiprocessing.cpu_count()
print(f"当前系统可用 CPU 核数: {cpu_count}")