<h2 dir="auto"><g-emoji class="g-emoji" alias="bug" fallback-src="https://github.githubassets.com/images/icons/emoji/unicode/1f41b.png">🐛</g-emoji> Bug</h2>
<p dir="auto">Search datasets...<br>
Original length: 900<br>
Offset: 0<br>
Limit: 900<br>
Final length: 900<br>
Search datasets...<br>
Original length: 120<br>
Offset: 0<br>
Limit: 120<br>
Final length: 120<br>
Using CuDNN in the experiment.<br>
Traceback (most recent call last):</p>
<p dir="auto">File "", line 1, in <br>
runfile('C:/Users/Administrator/Desktop/haoxiang_CRN/train.py', wdir='C:/Users/Administrator/Desktop/haoxiang_CRN')</p>
<p dir="auto">File "C:\Users\Administrator\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 827, in runfile<br>
execfile(filename, namespace)</p>
<p dir="auto">File "C:\Users\Administrator\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 110, in execfile<br>
exec(compile(f.read(), filename, 'exec'), namespace)</p>
<p dir="auto">File "C:/Users/Administrator/Desktop/haoxiang_CRN/train.py", line 103, in <br>
main(config, resume=resume)</p>
<p dir="auto">File "C:/Users/Administrator/Desktop/haoxiang_CRN/train.py", line 93, in main<br>
validation_dataloader=valid_data_loader</p>
<p dir="auto">File "C:\Users\Administrator\Desktop\haoxiang_CRN\trainer\trainer.py", line 27, in <strong>init</strong><br>
super(Trainer, self).<strong>init</strong>(config, resume, model, optimizer, loss_function)</p>
<p dir="auto">File "C:\Users\Administrator\Desktop\haoxiang_CRN\trainer\base_trainer.py", line 15, in <strong>init</strong><br>
self.model = model.to(self.device)</p>
<p dir="auto">File "C:\Users\Administrator\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 432, in to<br>
return self._apply(convert)</p>
<p dir="auto">File "C:\Users\Administrator\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 208, in _apply<br>
module._apply(fn)</p>
<p dir="auto">File "C:\Users\Administrator\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 230, in _apply<br>
param_applied = fn(param)</p>
<p dir="auto">File "C:\Users\Administrator\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 430, in convert<br>
return t.to(device, dtype if t.is_floating_point() else None, non_blocking)</p>
<p dir="auto">RuntimeError: CUDA error: unspecified launch failure</p>
<h2 dir="auto">Expected behavior</h2>
<p dir="auto">When I run on a small data set, the above error occurs when the data set becomes larger.</p>
<h2 dir="auto">Environment</h2>
<p dir="auto">Collecting environment information...<br>
PyTorch version: 1.2.0<br>
Is debug build: No<br>
CUDA used to build PyTorch: 10.0</p>
<p dir="auto">OS: Microsoft Windows 10 Professional<br>
GCC version: Could not collect<br>
CMake version: Could not collect</p>
<p dir="auto">Python version: 3.7<br>
Is CUDA available: Yes<br>
CUDA runtime version: 10.0.130<br>
GPU models and configuration: Could not collect<br>
Nvidia driver version: Could not collect<br>
cuDNN version: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin\cudnn64_7.dll</p>
<p dir="auto">Versions of relevant libraries:<br>
[pip] numpy==1.16.5<br>
[pip] numpydoc==0.9.1<br>
[pip] torch==1.2.0<br>
[pip] torchvision==0.4.0<br>
[conda] blas                      1.0                         mkl    defaults<br>
[conda] mkl                       2019.4                      245    defaults<br>
[conda] mkl-service               2.3.0            py37hb782905_0    defaults<br>
[conda] mkl_fft                   1.0.14           py37h14836fe_0    defaults<br>
[conda] mkl_random                1.1.0            py37h675688f_0    defaults<br>
[conda] torch                     1.2.0                    pypi_0    pypi<br>
[conda] torchvision               0.4.0                    pypi_0    pypi</p>
<h2 dir="auto">Additional context</h2>

<p dir="auto">win10[1903]<br>
Graphics card model：RTX2060<br>
anaconda 4.7.12<br>
Graphics driver 441.20<br>
CUDA:cuda_10.0.130_411.31_win10<br>
CUDNN:cudnn-10.0-windows10-x64-v7.6.0.64<br>
Python:3.7.4</p>
<p dir="auto">Do you have any good suggestions for this kind of problem?</p>
<h2 dir="auto"><g-emoji class="g-emoji" alias="bug" fallback-src="https://github.githubassets.com/images/icons/emoji/unicode/1f41b.png">🐛</g-emoji> Bug</h2>
<p dir="auto">No problem in PyTorch 1.2. Archive with code and data: <a href="https://github.com/pytorch/pytorch/files/3723821/PyTorch.zip">https://github.com/pytorch/pytorch/files/3723821/PyTorch.zip</a></p>
<p dir="auto">Windows 10 (1903), Python 3.7.4, RTX 2060 (driver version 436.48)</p>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
&lt;ipython-input-6-68308ed1e055&gt; in &lt;module&gt;
     35         cum_loss.append(loss.item())
     36 
---&gt; 37         loss.backward()
     38         optimizer.step()
     39 

C:\Anaconda3\envs\torch13\lib\site-packages\torch\tensor.py in backward(self, gradient, retain_graph, create_graph)
    148                 products. Defaults to ``False``.
    149         &quot;&quot;&quot;
--&gt; 150         torch.autograd.backward(self, gradient, retain_graph, create_graph)
    151 
    152     def register_hook(self, hook):

C:\Anaconda3\envs\torch13\lib\site-packages\torch\autograd\__init__.py in backward(tensors, grad_tensors, retain_graph, create_graph, grad_variables)
     97     Variable._execution_engine.run_backward(
     98         tensors, grad_tensors, retain_graph, create_graph,
---&gt; 99         allow_unreachable=True)  # allow_unreachable flag
    100 
    101 

RuntimeError: CUDA error: unspecified launch failure"><pre class="notranslate"><code class="notranslate">---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
&lt;ipython-input-6-68308ed1e055&gt; in &lt;module&gt;
     35         cum_loss.append(loss.item())
     36 
---&gt; 37         loss.backward()
     38         optimizer.step()
     39 

C:\Anaconda3\envs\torch13\lib\site-packages\torch\tensor.py in backward(self, gradient, retain_graph, create_graph)
    148                 products. Defaults to ``False``.
    149         """
--&gt; 150         torch.autograd.backward(self, gradient, retain_graph, create_graph)
    151 
    152     def register_hook(self, hook):

C:\Anaconda3\envs\torch13\lib\site-packages\torch\autograd\__init__.py in backward(tensors, grad_tensors, retain_graph, create_graph, grad_variables)
     97     Variable._execution_engine.run_backward(
     98         tensors, grad_tensors, retain_graph, create_graph,
---&gt; 99         allow_unreachable=True)  # allow_unreachable flag
    100 
    101 

RuntimeError: CUDA error: unspecified launch failure
</code></pre></div>
<p dir="auto">cc <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/ezyang/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/ezyang">@ezyang</a> <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/gchanan/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/gchanan">@gchanan</a> <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/zou3519/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/zou3519">@zou3519</a> <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/ssnl/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/ssnl">@ssnl</a> <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/albanD/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/albanD">@albanD</a> <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/gqchen/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/gqchen">@gqchen</a> <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/ngimel/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/ngimel">@ngimel</a> <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/peterjc123/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/peterjc123">@peterjc123</a></p>