解决方法:
造成这个错误的原因主要是GPU编号原因或者cuda安装版本不匹配
例如当本地只有一个 GPU (GPU:0),而程序中使用 GPUs:1。
因此,在程序中找到定义 device 的地方,我的如下:
device = th. device ("cuda:1")
将其中 “cuda:1” 修改为 “cuda:0”,程序将不在报错。
device = th. device ("cuda:0")
解决方法:
造成这个错误的原因主要是GPU编号原因或者cuda安装版本不匹配
例如当本地只有一个 GPU (GPU:0),而程序中使用 GPUs:1。
因此,在程序中找到定义 device 的地方,我的如下:
device = th. device ("cuda:1")
将其中 “cuda:1” 修改为 “cuda:0”,程序将不在报错。
device = th. device ("cuda:0")