Traceback (most recent call last):


 "<stdin>", line 1, in <module>


 "C:\Users\Jonas\Anaconda3\envs\schnetenv\lib\site-packages\torch_geometric\data\data.py", line 247, in to

return self.apply(lambda x: x.to(device), *keys)


 "C:\Users\Jonas\Anaconda3\envs\schnetenv\lib\site-packages\torch_geometric\data\data.py", line 233, in apply

self[key] = func(item)


 "C:\Users\Jonas\Anaconda3\envs\schnetenv\lib\site-packages\torch_geometric\data\data.py", line 247, in <lambda>

return self.apply(lambda x: x.to(device), *keys)


 "C:\Users\Jonas\Anaconda3\envs\schnetenv\lib\site-packages\torch\cuda\__init__.py", line 163, in _lazy_init

torch._C._cuda_init()
RuntimeError: CUDA error: unknown error

Traceback (most recent call last):


 "tools/train.py", line 95, in <module>

main()


 "tools/train.py", line 91, in main

logger=logger)


 "/home/pg/mmdetection/mmdet/apis/train.py", line 61, in train_detector

_non_dist_train(model, dataset, cfg, validate=validate)


 "/home/pg/mmdetection/mmdet/apis/train.py", line 185, in _non_dist_train

model = MMDataParallel(model, device_ids=range(cfg.gpus)).cuda()


 "/home/pg/anaconda3/envs/open-mmlab/lib/python3.7/site-packages/torch/nn/parallel/data_parallel.py", line 136, in __init__

self.module.cuda(device_ids[0])


 "/home/pg/anaconda3/envs/open-mmlab/lib/python3.7/site-packages/torch/nn/modules/module.py", line 265, in cuda

return self._apply(lambda t: t.cuda(device))


 "/home/pg/anaconda3/envs/open-mmlab/lib/python3.7/site-packages/torch/nn/modules/module.py", line 193, in _apply

module._apply(fn)


 "/home/pg/anaconda3/envs/open-mmlab/lib/python3.7/site-packages/torch/nn/modules/module.py", line 193, in _apply

module._apply(fn)


 "/home/pg/anaconda3/envs/open-mmlab/lib/python3.7/site-packages/torch/nn/modules/module.py", line 199, in _apply

param.data = fn(param.data)


 "/home/pg/anaconda3/envs/open-mmlab/lib/python3.7/site-packages/torch/nn/modules/module.py", line 265, in <lambda>

return self._apply(lambda t: t.cuda(device))
RuntimeError: CUDA error: unknown error
