日本免费高清视频-国产福利视频导航-黄色在线播放国产-天天操天天操天天操天天操|www.shdianci.com

學無先后，達者為師

網站首頁 編程語言 正文

RuntimeError: CUDA error: invalid device ordinal

作者：香博士 更新時間： 2022-02-18 編程語言

在這里插入圖片描述
解決方法：
造成這個錯誤的原因主要是GPU編號原因或者cuda安裝版本不匹配

例如當本地只有一個 GPU (GPU:0)，而程序中使用 GPUs:1。

因此，在程序中找到定義 device 的地方，我的如下：

device = th. device ("cuda:1")

將其中 “cuda:1” 修改為 “cuda:0”，程序將不在報錯。

device = th. device ("cuda:0")

原文鏈接：https://blog.csdn.net/weixin_42535423/article/details/109320965

欄目分類
最近更新