code: def test_get_single_device(driver, device):
      """
      测试正常情况下初始化TorchSingleDriver的情况
      """
  
      model = TorchNormalModel_Classification_1(20, 10)
      driver = initialize_torch_driver(driver, device, model)
      assert isinstance(driver, TorchSingleDriver)
  
file_path: fastNLP/tests/core/drivers/torch_driver/test_initialize_torch_driver.py
imports: import pytest
  from fastNLP.core.drivers import TorchSingleDriver, TorchDDPDriver
  from fastNLP.core.drivers.torch_driver.initialize_torch_driver import initialize_torch_driver
  from tests.helpers.models.torch_model import TorchNormalModel_Classification_1
  from tests.helpers.utils import magic_argv_env_context
  from fastNLP.envs.imports import _NEED_IMPORT_TORCH
label_desc: Use image node_id 2 for calling a global function with example usage: test_get_single_device(driver, device) without return types