GPU的使用问题

[Gavin4th]

想问下作者，为啥我能运行程序，但是却是使用CPU运行的

[hy-chen]

想问下作者，为啥我能运行程序，但是却是使用CPU运行的

看看你的output有没有这一行：failed call to cuInit: CUDA_ERROR_NO_DEVICE: no CUDA-capable device is detected 因为在run.py里面它设置了cuda_device = 1.但是大部分人都只有1个cudadevice，把1改成0就好了 os.environ["CUDA_VISIBLE_DEVICES"] = "0"

[fenss]

run.py里面第6行代码是：

os.environ["CUDA_VISIBLE_DEVICES"] = "1"

作者应该是在多卡环境下跑的代码，可能你是单机，所以“1”改成“0”就可以了

[Gavin4th]

谢谢，我试试

------------------ 原始邮件 ------------------ 发件人: "fenss"<[email protected]>; 发送时间: 2021年1月8日(星期五) 上午9:05 收件人: "weizhepei/CasRel"<[email protected]>; 抄送: "Jarvan"<[email protected]>; "Author"<[email protected]>; 主题: Re: [weizhepei/CasRel] GPU的使用问题 (#40)

run.py里面第6行代码是：

os.environ["CUDA_VISIBLE_DEVICES"] = "1" 作者应该是在多卡环境下跑的代码，可能你是单机，所以“1”改成“0”就可以了

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.