如何正确运行我的项目使用 GPU？

Question

我对 Torch 和机器学习真的很陌生。我正在尝试运行我的项目使用 GPU。我试图对我的代码进行这样的修改：

model = Challenge()
    model = model.to(torch.device('cuda'))

但是，我仍然遇到以下错误：

Traceback (most recent call last):
  File "C:/Users/ruidong/Desktop/YZR temp/Project2/train_challenge.py", line 112, in <module>
    main()
  File "C:/Users/ruidong/Desktop/YZR temp/Project2/train_challenge.py", line 91, in main
    stats)
  File "C:/Users/ruidong/Desktop/YZR temp/Project2/train_challenge.py", line 40, in _evaluate_epoch
    output = model(X)
  File "C:\Users\ruidong\Anaconda3\envs\EECS445\lib\site-packages\torch\nn\modules\module.py", line 532, in __call__
    result = self.forward(*input, **kwargs)
  File "C:\Users\ruidong\Desktop\YZR temp\Project2\model\challenge.py", line 48, in forward
    z = F.relu(self.conv1(x))
  File "C:\Users\ruidong\Anaconda3\envs\EECS445\lib\site-packages\torch\nn\modules\module.py", line 532, in __call__
    result = self.forward(*input, **kwargs)
  File "C:\Users\ruidong\Anaconda3\envs\EECS445\lib\site-packages\torch\nn\modules\conv.py", line 345, in forward
    return self.conv2d_forward(input, self.weight)
  File "C:\Users\ruidong\Anaconda3\envs\EECS445\lib\site-packages\torch\nn\modules\conv.py", line 342, in conv2d_forward
    self.padding, self.dilation, self.groups)
RuntimeError: Expected object of device type cuda but got device type cpu for argument #1 'self' in call to _thnn_conv2d_forward

有什么建议吗？非常感谢。

Answer 1

模型已正确移动到 GPU。但是，对于放置在 GPU 中的模型，您还需要传递 GPU 中的张量。该错误是因为您传递的张量位于 GPU 模型中的 CPU 中。在将输入传递给模型

之前，只需对输入执行相同的操作

如何正确运行我的项目使用 GPU？

How to properly run my project using GPU?

python

gpu

torch

pytorch

如何正确 运行 我的项目使用 GPU？

How to properly run my project using GPU?

python

gpu

torch

pytorch

如何正确运行我的项目使用 GPU？