是否支持 windows 环境 + cpu运行

[lishuaijun]

请问我可以在windows + cpu 的环境下运行此项目吗？
目前出现了因 PyTorch库没有编译支持CUDA 引发的报错，但是实际上我的pytorch是使用的 cpu版本，
即：conda install pytorch torchvision torchaudio cpuonly -c pytorch

目前调整了部分代码，
如 ：# self.pfld_backbone = PFLDInference().cuda() 这里改成了cpu
self.pfld_backbone = PFLDInference().to('cpu')

    # input = input.cuda() 这里改成了cpu
    input = input.cpu()

   # checkpoint = torch.load('./checkpoint_epoch_335.pth.tar') 这里改成了cpu
    checkpoint = torch.load('./checkpoint_epoch_335.pth.tar', map_location=torch.device('cpu'), weights_only=True)

但依旧还有其他报错

[Doonkey]

将.cuda ()删除就可以在cpu 上跑了，只是慢，而且是很慢

…

---- 回复的原邮件 ---- | 发件人 | ***@***.***> | | 日期 | 2024年12月10日 11:49 | | 收件人 | ***@***.***> | | 抄送至 | ***@***.***> | | 主题 | [anliyuan/Ultralight-Digital-Human] 是否支持 windows 环境 + cpu运行 (Issue #98) | 请问我可以在windows + cpu 的环境下运行此项目吗？ 目前出现了因 PyTorch库没有编译支持CUDA 引发的报错，但是实际上我的pytorch是使用的 cpu版本， 即：conda install pytorch torchvision torchaudio cpuonly -c pytorch 目前调整了部分代码， 如 ：# self.pfld_backbone = PFLDInference().cuda() 这里改成了cpu self.pfld_backbone = PFLDInference().to('cpu') # input = input.cuda() 这里改成了cpu input = input.cpu() # checkpoint = torch.load('./checkpoint_epoch_335.pth.tar') 这里改成了cpu checkpoint = torch.load('./checkpoint_epoch_335.pth.tar', map_location=torch.device('cpu'), weights_only=True) 但依旧还有其他报错 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: ***@***.***>

[lishuaijun]

将.cuda()删除就可以在cpu上跑了，只是慢，而且是很慢
…
---- 回复的原邮件 ---- | 发件人 | @.> | | 日期 | 2024年12月10日 11:49 | | 收件人 | @.> | | 抄送至 | @.> | | 主题 | [anliyuan/Ultralight-Digital-Human] 是否支持 windows 环境 + cpu运行 (Issue #98) | 请问我可以在windows + cpu 的环境下运行此项目吗？ 目前出现了因 PyTorch库没有编译支持CUDA 引发的报错，但是实际上我的pytorch是使用的 cpu版本， 即：conda install pytorch torchvision torchaudio cpuonly -c pytorch 目前调整了部分代码， 如 ：# self.pfld_backbone = PFLDInference().cuda() 这里改成了cpu self.pfld_backbone = PFLDInference().to('cpu') # input = input.cuda() 这里改成了cpu input = input.cpu() # checkpoint = torch.load('./checkpoint_epoch_335.pth.tar') 这里改成了cpu checkpoint = torch.load('./checkpoint_epoch_335.pth.tar', map_location=torch.device('cpu'), weights_only=True) 但依旧还有其他报错 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: @.>
我看到有很多处类似的 .cuda() , 比如 get_landmark.py hubert.py 以及下层引用中均有，是删掉外层的就可以了吗？（不是专业python人士，问的问题可能有些业余T_T）

[Doonkey]

批量替换.cuda()为空字符串即可，这里还是如你写的一样需要指定cpu. torch.load('./checkpoint_epoch_335.pth.tar', map_location=torch.device('cpu')) | | doonkey | | ***@***.*** | ---- 回复的原邮件 ---- | 发件人 | ***@***.***> | | 发送日期 | 2024年12月11日 15:37 | | 收件人 | ***@***.***> | | 抄送人 | ***@***.***> , ***@***.***> | | 主题 | Re: [anliyuan/Ultralight-Digital-Human] 是否支持 windows 环境 + cpu运行 (Issue #98) | 将.cuda()删除就可以在cpu上跑了，只是慢，而且是很慢 … ---- 回复的原邮件 ---- | 发件人 | @.> | | 日期 | 2024年12月10日 11:49 | | 收件人 | @.> | | 抄送至 | @.> | | 主题 | [anliyuan/Ultralight-Digital-Human] 是否支持 windows 环境 + cpu运行 (Issue #98) | 请问我可以在windows + cpu 的环境下运行此项目吗？ 目前出现了因 PyTorch库没有编译支持CUDA 引发的报错，但是实际上我的pytorch是使用的 cpu版本， 即：conda install pytorch torchvision torchaudio cpuonly -c pytorch 目前调整了部分代码， 如 ：# self.pfld_backbone = PFLDInference().cuda() 这里改成了cpu self.pfld_backbone = PFLDInference().to('cpu') # input = input.cuda() 这里改成了cpu input = input.cpu() # checkpoint = torch.load('./checkpoint_epoch_335.pth.tar') 这里改成了cpu checkpoint = torch.load('./checkpoint_epoch_335.pth.tar', map_location=torch.device('cpu'), weights_only=True) 但依旧还有其他报错 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: @.> 我看到有很多处类似的 .cuda() , 比如 get_landmark.py hubert.py 以及下层引用中均有，是删掉外层的就可以了吗？（不是专业python人士，问的问题可能有些业余T_T） — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: ***@***.***>

[lishuaijun]

批量替换.cuda()为空字符串即可，这里还是如你写的一样需要指定cpu. torch.load('./checkpoint_epoch_335.pth.tar', map_location=torch.device('cpu')) | | doonkey | | @.*** | ---- 回复的原邮件 ---- | 发件人 | @.> | | 发送日期 | 2024年12月11日 15:37 | | 收件人 | @.> | | 抄送人 | @.> , @.> | | 主题 | Re: [anliyuan/Ultralight-Digital-Human] 是否支持 windows 环境 + cpu运行 (Issue #98) | 将.cuda()删除就可以在cpu上跑了，只是慢，而且是很慢 … ---- 回复的原邮件 ---- | 发件人 | @.> | | 日期 | 2024年12月10日 11:49 | | 收件人 | @.> | | 抄送至 | @.> | | 主题 | [anliyuan/Ultralight-Digital-Human] 是否支持 windows 环境 + cpu运行 (Issue #98) | 请问我可以在windows + cpu 的环境下运行此项目吗？ 目前出现了因 PyTorch库没有编译支持CUDA 引发的报错，但是实际上我的pytorch是使用的 cpu版本， 即：conda install pytorch torchvision torchaudio cpuonly -c pytorch 目前调整了部分代码， 如 ：# self.pfld_backbone = PFLDInference().cuda() 这里改成了cpu self.pfld_backbone = PFLDInference().to('cpu') # input = input.cuda() 这里改成了cpu input = input.cpu() # checkpoint = torch.load('./checkpoint_epoch_335.pth.tar') 这里改成了cpu checkpoint = torch.load('./checkpoint_epoch_335.pth.tar', map_location=torch.device('cpu'), weights_only=True) 但依旧还有其他报错 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: @.> 我看到有很多处类似的 .cuda() , 比如 get_landmark.py hubert.py 以及下层引用中均有，是删掉外层的就可以了吗？（不是专业python人士，问的问题可能有些业余T_T） — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you commented.Message ID: @.***>

感谢 我来试试