使用web camera出現 Expected object of backend CPU but got backend CUDA for argument - Cupoy

2020/03/29 下午 11:06

電腦視覺深度學習討論版

周彥仰

觀看數：6

回答數：2

收藏數：0

cvdl-2

您好, 我想使用web camera測試效果, 但在執行程式後, 會顯示下列的錯誤資訊

RuntimeError: Expected object of backend CPU but got backend CUDA for argument #2 'weight'

是否因為我是用gpu版, 而有部分程式不相容呢?

回答列表

• 2020/03/30 上午 09:29
  Jeffrey
  贊同數：1
  不贊同數：0
  留言數：0

  你好, 請問一下, 是否有指定 GPU 或是 CPU 的device? 

• 2020/03/31 下午 06:00
  楊哲寧
  贊同數：1
  不贊同數：0
  留言數：1

  您好，這個部分是模型有轉成cuda模式，但是input_tensor沒有，可以將input_tensor.type(torch.cuda.FloatTensor)轉過去