Build SSD_VGG.ipynb GPU執行問題 - Cupoy

2020/04/15 下午 10:14

電腦視覺深度學習討論版

呂季修

觀看數：7

回答數：2

收藏數：0

VGG

我使用colab執行Build SSD_VGG.ipynb到最後出現這個問題 想請問怎麼解決這個問題

回答列表

• 2020/04/16 下午 03:43
  Jeffrey
  贊同數：0
  不贊同數：0
  留言數：1

  請問一下, 需先確認兩件事:

  1. CUDA GPU  support, 並且有 得到使用

  2. 相關的檔案都存在?

• 2020/04/22 上午 11:57
  楊哲寧
  贊同數：0
  不贊同數：0
  留言數：3

  您好，上面那個只是他的warning可以忽略，真正錯誤在於out跟y其中有一個是cpu type(應該是y)，可以簡單將tensor轉成cuda即可，ex 多一行 y.type(torch.cuda.LongTensor)