CNML: 7.10.2 0a592c0

CNRT: 4.10.1 a884a9a

CNML: 7.10.2 0a592c0

CNRT: 4.10.1 a884a9a

Traceback (most recent call last):

  File "main_imagenet.py", line 84, in

    torch.cuda.set_device(args.gpu)

  File "/workspace/volume/private/sdk/venv/pytorch/lib/python3.7/site-packages/torch/cuda/__init__.py", line 300, in set_device

    torch._C._cuda_setDevice(device)

AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'

Params: 12.85M, Multi-adds: 2040.803M

Traceback (most recent call last):

  File "main_imagenet.py", line 84, in

    torch.cuda.set_device(args.gpu)

  File "/workspace/volume/private/sdk/venv/pytorch/lib/python3.7/site-packages/torch/cuda/__init__.py", line 300, in set_device

    torch._C._cuda_setDevice(device)

AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'

Traceback (most recent call last):

  File "/usr/local/lib/python3.7/runpy.py", line 193, in _run_module_as_main

    "__main__", mod_spec)

  File "/usr/local/lib/python3.7/runpy.py", line 85, in _run_code

    exec(code, run_globals)

  File "/workspace/volume/private/sdk/venv/pytorch/lib/python3.7/site-packages/torch/distributed/launch.py", line 253, in

    main()

  File "/workspace/volume/private/sdk/venv/pytorch/lib/python3.7/site-packages/torch/distributed/launch.py", line 249, in main

    cmd=cmd)

subprocess.CalledProcessError: Command '['/workspace/volume/private/sdk/venv/pytorch/bin/python', '-u', 'main_imagenet.py', '--local_rank=1', '--arch', 'HCGNet_B', '--gpu-id', '0,1', '--resume', '--checkpoint', '../models/HCGNet_B_best.pth.tar', '--evaluate']' returned non-zero exit status 1.