test_environment.py 文件源码

def test_nogpu(self):
    cupy = None
    if hasattr(chainer.cuda, "cupy"):
      cupy = getattr(chainer.cuda, "cupy")
      delattr(chainer.cuda, "cupy")
    environment.init_gpu(0)
    self.assertEqual(environment.array_module(), numpy)
    if cupy is not None:
      setattr(chainer.cuda, "cupy", cupy)