PyTorch训练网络时出现下列错误
cuda runtime error (4) : unspecified launch failure at c:\anaconda2\conda...
解决方法一(亲测有效):
“开始”
菜单→NVIDIA Corporation
(文件夹),进入Nsight Monitor
- 点开右下角的
options
选项 - 找到Microsoft display driver下的
WDDM TDR enable
,设置为False,因为该选项导致kernel运行时间超过指定值会中断。
step 1~3
:
![](https://i-blog.csdnimg.cn/blog_migrate/74710051970147246d388baa33c518ec.png)
![](https://i-blog.csdnimg.cn/blog_migrate/be9a1432fbaf51ad2a75e93ae40cd48e.png)
![](https://i-blog.csdnimg.cn/blog_migrate/ab0d8f5215e47fcbc8fb12b0b3450729.png)
解决方法二(未尝试):
出处:
https://github.com/tensorflow/tensorflow/issues/33536
解决办法:降低Nvidia驱动版本,比如 Nvidia 436.** 降至 431.60