关于报错AssertionError: CUDA unavailable, invalid device 0 requested的处理方式

    从git上找了一个工程,按照其readme.md想运行试试,遇到报错:

关于报错AssertionError: CUDA unavailable, invalid device 0 requested的处理方式_第1张图片

这个报错是指:代码要运行CUDA,但是不能运行。

解决办法分两种情况:1.不运行CUDA;2.运行CUDA。

针对情况1:

点击蓝色 .\torch_utils.py 文件,传送到下图:

关于报错AssertionError: CUDA unavailable, invalid device 0 requested的处理方式_第2张图片

 继续按照其readme.md运行试试,会遇到如下报错:

关于报错AssertionError: CUDA unavailable, invalid device 0 requested的处理方式_第3张图片

此时,点击报错中倒数第二个upsampling.py文件, 传送到下图:

关于报错AssertionError: CUDA unavailable, invalid device 0 requested的处理方式_第4张图片

屏蔽后如下图:

到此,不运行CUDA的情况下可以正常运行程序了。

 针对情况2:

可参照下面博客:AssertionError: CUDA unavailable, invalid device 0 requested_消灭BUG鸭的博客-CSDN博客_cuda unavailable

你可能感兴趣的:(python,opencv,pytorch,图像处理)