TadaoYamaoka / DeepLearningShogi

GNU General Public License v3.0
201 stars 33 forks source link

GPU ID 0以外を使用するデータローダ並列化でエラーが発生する #40

Closed TadaoYamaoka closed 3 years ago

TadaoYamaoka commented 3 years ago

Traceback (most recent call last): File "C:\Anaconda3\lib\runpy.py", line 194, in _run_module_as_main return _run_code(code, main_globals, None, File "C:\Anaconda3\lib\runpy.py", line 87, in _run_code exec(code, run_globals) File "c:\users\kei\src\deeplearningshogi\dlshogi\train_rl_policy_with_value_using_hcpe_bootstrap.py", line 210, in y1, y2 = model(x1, x2) File "c:\users\kei\src\deeplearningshogi\dlshogi\policy_value_network_resnet10_swish.py", line 81, in call u1_1_1 = self.l1_1_1(x1) File "C:\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl result = self.forward(*input, **kwargs) File "C:\Anaconda3\lib\site-packages\torch\nn\modules\conv.py", line 423, in forward return self._conv_forward(input, self.weight) File "C:\Anaconda3\lib\site-packages\torch\nn\modules\conv.py", line 419, in _conv_forward return F.conv2d(input, weight, self.bias, self.stride, RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 0 does not equal 1 (while checking arguments for cudnn_convolution)