RuntimeError: CUDA error: an illegal memory access was encountered

[Jiaxi-Z]

你好lee-zq，

非常感谢你的方法！ 我在训练的时候遇到了这个问题:

=======Epoch:1=======
  0%|                                                                                                                                                                                | 0/2 [00:00<?, ?it/s]/home/jiaxi/anaconda3/envs/DKFZ/lib/python3.6/site-packages/scipy/ndimage/interpolation.py:583: UserWarning: From scipy 0.13.0, the output shape of zoom() is calculated with round() instead of int() - for these inputs the size of the returned array has changed.
  "the returned array has changed.", UserWarning)
  0%|                                                                                                                                                                                | 0/2 [00:04<?, ?it/s]
Traceback (most recent call last):
  File "train.py", line 103, in <module>
    train_log = train(model, train_loader, optimizer)
  File "train.py", line 66, in train
    optimizer.step()
  File "/home/jiaxi/anaconda3/envs/DKFZ/lib/python3.6/site-packages/torch/autograd/grad_mode.py", line 26, in decorate_context
    return func(*args, **kwargs)
  File "/home/jiaxi/anaconda3/envs/DKFZ/lib/python3.6/site-packages/torch/optim/adam.py", line 119, in step
    group['eps']
  File "/home/jiaxi/anaconda3/envs/DKFZ/lib/python3.6/site-packages/torch/optim/functional.py", line 94, in adam
    denom = (exp_avg_sq.sqrt() / math.sqrt(bias_correction2)).add_(eps)
RuntimeError: CUDA error: an illegal memory access was encountered

非常感谢你的帮助。

[Jiaxi-Z]

你好，我之前遇到的问题是发生在Ubuntu 20.04 上的，虽然不知道为什么。 切换为Windows10，就没有这个问题了。