WindowsでPytorchのCycleGANを使うことができません

前提・実現したいこと

あるサイトのページ(URL:https://touch-sp.hatenablog.com/entry/2018/10/20/215105)を参考にしてWindowsでCycleGANを実装しようとしました。
しかし、公開データセットで実装まではできたのですが、実行の際に
下記のエラーが出てしまい、実行できませんでした。

ご教授お願いいたします。

発生している問題・エラーメッセージ

（Traceback (most recent call last):
  File "test.py", line 38, in <module>
    opt = TestOptions().parse()  # get test options
  File "C:\GAN\pytorch-CycleGAN-and-pix2pix-master\options\base_options.py", line 133, in parse
    torch.cuda.set_device(opt.gpu_ids[0])
  File "C:\ProgramData\Anaconda3\lib\site-packages\torch\cuda\__init__.py", line 292, in set_device
    torch._C._cuda_setDevice(device)
AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'）

＃申し訳ありません。import torch print(torch.cuda.is_available())でTUREになったのですが下記のようなエラーが発生しました。

RuntimeError: Error(s) in loading state_dict for ResnetGenerator:
        Missing key(s) in state_dict: "model.10.conv_block.6.weight", "model.10.conv_block.6.bias", "model.11.conv_block.6.weight", "model.11.conv_block.6.bias", "model.12.conv_block.6.weight", "model.12.conv_block.6.bias", "model.13.conv_block.6.weight", "model.13.conv_block.6.bias", "model.14.conv_block.6.weight", "model.14.conv_block.6.bias", "model.15.conv_block.6.weight", "model.15.conv_block.6.bias", "model.16.conv_block.6.weight", "model.16.conv_block.6.bias", "model.17.conv_block.6.weight", "model.17.conv_block.6.bias", "model.18.conv_block.6.weight", "model.18.conv_block.6.bias".
        Unexpected key(s) in state_dict: "model.10.conv_block.5.weight", "model.10.conv_block.5.bias", "model.10.conv_block.6.running_mean", "model.10.conv_block.6.running_var", "model.10.conv_block.6.num_batches_tracked", "model.11.conv_block.5.weight", "model.11.conv_block.5.bias", "model.11.conv_block.6.running_mean", "model.11.conv_block.6.running_var", "model.11.conv_block.6.num_batches_tracked", "model.12.conv_block.5.weight", "model.12.conv_block.5.bias", "model.12.conv_block.6.running_mean", "model.12.conv_block.6.running_var", "model.12.conv_block.6.num_batches_tracked", "model.13.conv_block.5.weight", "model.13.conv_block.5.bias", "model.13.conv_block.6.running_mean", "model.13.conv_block.6.running_var", "model.13.conv_block.6.num_batches_tracked", "model.14.conv_block.5.weight", "model.14.conv_block.5.bias", "model.14.conv_block.6.running_mean", "model.14.conv_block.6.running_var", "model.14.conv_block.6.num_batches_tracked", "model.15.conv_block.5.weight", "model.15.conv_block.5.bias", "model.15.conv_block.6.running_mean", "model.15.conv_block.6.running_var", "model.15.conv_block.6.num_batches_tracked", "model.16.conv_block.5.weight", "model.16.conv_block.5.bias", "model.16.conv_block.6.running_mean", "model.16.conv_block.6.running_var", "model.16.conv_block.6.num_batches_tracked", "model.17.conv_block.5.weight", "model.17.conv_block.5.bias", "model.17.conv_block.6.running_mean", "model.17.conv_block.6.running_var", "model.17.conv_block.6.num_batches_tracked", "model.18.conv_block.5.weight", "model.18.conv_block.5.bias", "model.18.conv_block.6.running_mean", "model.18.conv_block.6.running_var", "model.18.conv_block.6.num_batches_tracked".

該当のソースコード

python test.py --dataroot horse2zebra/testA --name horse2zebra_pretrained --model test

試したこと

他のサイト複数を見て試してみましたが、同様のエラーが発生しました。

補足情報（FW/ツールのバージョンなど）

Pythonのバージョンは3.8.2
Windows10 Pro 64bit
NVIDIA GeForce GTX1080
CUDA9.0
cudnn7.6.5
Python3.8.5 です。

jbpb0

2021/02/20 20:35

Pythonで下記を実行したら、どうなりますか？ import torch print(torch.cuda.is_available())

KMRZR

2021/02/20 23:12

falseになっておりました。申し訳ありませんが、解決法をご教授お願いします。

jbpb0

2021/02/21 04:34

GPU版PyTorchのインストールがちゃんとできてないようです https://pytorch.org/get-started/locally/ https://pytorch.org/get-started/previous-versions/ から、現在お使いのPyTorchとCUDAの両方のバージョンに合ったインストールコマンドを調べてください実際にPyTorchのインストールの際に使用したコマンドは、上記と合ってましたでしょうか？なお、 > CUDA9.0 に正式に対応したPyTorchのバージョンを上記Webページから探すと、一番新しいので1.1.0ですねもっと新しいバージョンのPyTorchをお使いなら、CUDAのバージョンを新しくした方がいいかも

KMRZR

2021/02/23 12:46

数日粘って import torch print(torch.cuda.is_available())でTrueが出るようになりましたが、今度は上記のようなエラーが発生しました。申し訳ありませんが、解決法をご教授お願いします。

行動規範の内容に同意します

回答1件

ベストアンサー

AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'）

GPU版PyTorchのインストールがちゃんとできてないようです

python
1import torch
2print(torch.cuda.is_available())

を実行して「True」にならないと、GPUが使えません

START LOCALLY
INSTALLING PREVIOUS VERSIONS OF PYTORCH
から、現在お使いのPyTorchとCUDAの両方のバージョンに合ったインストールコマンドを調べて、実際にPyTorchのインストールの際に使用したコマンドがそれと合ってたか確認してみてください

投稿2021/02/28 09:52

jbpb0

総合スコア7653