首页 > 编程知识 正文

invalid device ordinal,cudagetdevice报错

时间:2023-05-03 08:04:05 阅读:116111 作者:91

机器上有八张显卡,占用了两三张。 选择空闲的GPU后仍报告错误。

使用以下代码检查可用的GPU时,n_gpu为1,显示问号

n_gpu=torch.cuda.device_count ) )解析:在主文件的开头加上此代码,再次查看可用的GPU时为8

import osos.environ [ ' cuda _ visible _ devices ' ]=' 0,1,2,3,4,5,6,7 '

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。