AssertionError: Invalid device id GPU 가능합니다.

Question

docker 컨테이너에서 pytorch로 object detection training 중입니다.사용중인 깃허브: https://github.com/Banconxuan/RTM3D#monocular-3d-detection-with-geometric-constraints-embedding-and-semi-supervised-training-km3d Using tensorboardX Fix size testing. training chunk_sizes: [16, 16] The output will be saved to ./kitti_format/exp/KM3D_res18 heads {'hm': 3, 'wh': 2, 'hps': 18, 'rot': 8, 'dim': 3, 'prob': 1, 'reg': 2, 'hm_hp': 9, 'hp_offset': 2} Namespace(K=100, aggr_weight=0.0, agnostic_ex=False, arch='res_18', aug_ddd=0.5, aug_rot=0, batch_size=32, calib_dir='', cat_spec_wh=False, center_thresh=0.1, chunk_sizes=[16, 16], coor_thresh=0.3, data_dir='./kitti_format/data', dataset='kitti', debug=0, debug_dir='./kitti_format/exp/KM3D_res18/debug', debugger_theme='white', demo='', dense_hp=False, dense_wh=False, dim_weight=2, distribute=False, down_ratio=4, eval_oracle_dep=False, eval_oracle_hm=False, eval_oracle_hmhp=False, eval_oracle_hp_offset=False, eval_oracle_kps=False, eval_oracle_offset=False, eval_oracle_wh=False, exp_dir='./kitti_format/exp', exp_id='KM3D_res18', fix_res=True, flip=0.5, flip_idx=[[0, 1], [2, 3], [4, 5], [6, 7]], flip_test=False, gpus=[0, 1], gpus_str='0,1', head_conv=64, heads={'hm': 3, 'wh': 2, 'hps': 18, 'rot': 8, 'dim': 3, 'prob': 1, 'reg': 2, 'hm_hp': 9, 'hp_offset': 2}, hide_data_time=False, hm_hp=True, hm_hp_weight=1, hm_weight=1, hp_weight=1, input_h=384, input_res=1280, input_w=1280, keep_res=False, kitti_split='3dop', load_model='', logger_save=0, lr=0.000125, lr_step=[90, 120], master_batch_size=16, mean=array([[[0.485, 0.456, 0.406]]], dtype=float32), metric='loss', mse_loss=False, nms=False, no_color_aug=False, norm_wh=False, not_cuda_benchmark=False, not_hm_hp=False, not_prefetch_test=False, not_rand_crop=False, not_reg_bbox=False, not_reg_hp_offset=False, not_reg_offset=False, num_classes=3, num_epochs=200, num_iters=-1, num_stacks=1, num_workers=4, off_weight=1, output_h=96, output_res=320, output_w=320, pad=31, peak_thresh=0.2, print_iter=0, prob_weight=1, rect_mask=False, reg_bbox=True, reg_hp_offset=True, reg_loss='l1', reg_offset=True, results_dir='./kitti_format/exp/results', resume=False, root_dir='/workspace/RTM3D/src/lib/../..', rot_weight=0.2, rotate=0, save_all=False, save_dir='./kitti_format/exp/KM3D_res18', scale=0.4, scores_thresh=0.1, seed=317, shift=0.1, std=array([[[0.229, 0.224, 0.225]]], dtype=float32), stereo_aug=False, test=False, test_scales=[1.0], trainval=False, val_intervals=5, vis=False, vis_thresh=0.3, wh_weight=0.1) Creating model... => loading pretrained model https://download.pytorch.org/models/resnet18-5c106cde.pth Traceback (most recent call last): File "./src/main.py", line 114, in main(opt) File "./src/main.py", line 46, in main trainer.set_device(opt.gpus, opt.chunk_sizes, opt.device, opt.distribute) File "/workspace/RTM3D/src/lib/trains/base_trainer.py", line 60, in set_device chunk_sizes=chunk_sizes).to(device) File "/workspace/RTM3D/src/lib/models/data_parallel.py", line 127, in DataParallel return torch.nn.DataParallel(module, device_ids, output_device, dim) File "/opt/conda/envs/KM3D/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 131, in __init__ _check_balance(self.device_ids) File "/opt/conda/envs/KM3D/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 18, in _check_balance dev_props = [torch.cuda.get_device_properties(i) for i in device_ids] File "/opt/conda/envs/KM3D/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 18, in dev_props = [torch.cuda.get_device_properties(i) for i in device_ids] File "/opt/conda/envs/KM3D/lib/python3.6/site-packages/torch/cuda/__init__.py", line 301, in get_device_properties raise AssertionError("Invalid device id") AssertionError: Invalid device id

AssertionError: Invalid device id GPU 가능합니다.

0개의 답변

#GPU 목록