Python,CUDA環境で行われている演算をCPUに行わせたい

Question

GPUで行われている演算処理をCPUに行わせたい

Pramook Khungurn氏のソフトウェア「Talking Head Anime from a Single Image」
ホームページ
GitHub
を動かしてみたいのですが、Windows10の環境にNVIDIAのGPUがないためCUDAを扱えません。
そのため処理をCPUに行わせたいと考えています。

manual_poser.pyの

torch.device('cuda')
↓
torch.device('cpu')

としたところ以下のようなエラーが得られました

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False.
 If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

ログより、util.pyの

def torch_load(file_name):
    with open(file_name, 'rb') as f:
        return torch.load(f)

という記述部からCUDAを実行させるように見えたのですが、そこから何をすればいいかわからず困っています。

素人質問、さらにはぶん投げるような質問になってしまっていることは重々承知ですが、
行き詰ってしまっており、ご教授いただければと思います。

Comments

[episteme]

エラーメッセージの指示に従うだけじゃない？

[tiitoi]

動かしてないのでわかりませんが、torch.load(f, torch.device('cpu')) としてみてはどうでしょうか。
https://pytorch.org/docs/stable/torch.html#torch.load
https://github.com/explosion/spacy-transformers/issues/33

[tk4869]

GUIが起動し、処理を行うタイミングで

DefaultCPUAllocator: not enough memory: you tried to allocate ○○○○ bytes. Buy new RAM!

というエラーがでるようになりました。メモリの異常使用か単純なメモリ不足か判断がつかないためPCを掃除してから追記します

[tiitoi]

メモリは何GBでしょうか？
モデルにもよりますが、CNN を動かすには空きが8GBぐらいほしいところです。
なので、OS やその他アプリケーションで5GBぐらいはメモリが使われていると思うので、搭載メモリ量が16GBはないと、単純にメモリ不足ということかもしれません。

> PCを掃除してから追記します

HDD の容量は関係ないので、掃除が不要なファイルを消すということを指しているのであれば、意味がないです。
エラーは PC のメモリ量のことについて言っているので、足りないのだとすると、「Buy new RAM!」と記載があるように、メモリを増設するしかないです。

Answer 1

GUIによる出力は
manual_poser.pyの

torch.device('cuda')
↓
torch.device('cpu')

util.pyの

def torch_load(file_name):
    with open(file_name, 'rb') as f:
        return torch.load(f)
を
def torch_load(file_name):
    with open(file_name, 'rb') as f:
        return torch.load(f, torch.device('cpu')) 

とすることで出力されるようになりました。

webcamを用いた出力機能は試していませんがCPUの処理速度的に機能しないものと思われます。

皆さん本当にありがとうございました。