CUDA指定显卡时报错及解决-RuntimeError: Expected tensor for argument #1 ‘input’ to have the same device as tensor for argument #2 ‘weight’; but device 0 does not equal 1 (while checking arguments for cudnn_convolution)

最新文章

Python
CUDA指定显卡时报错及解决-RuntimeError: Expected tensor for argument #1 ‘input’ to have the same device as tensor for argument #2 ‘weight’; but device 0 does not equal 1 (while checking arguments for cudnn_convolution)
已阅： 284 2022/4/2：修改了题目描述，加入了报错信息错误信息RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 0 does not equal 1 (while checking arguments for cudnn_convolution) 还有一个错误信息找不到了。场景描述工具：pytorch, CUDA 需求：对于8块GPU，使用4号GPU进行训练。配置1os.e……
MWHLS 2022/03/18
0