Pytorchx のトレーニング中に、GPU マルチスレッドを使用して画像を読み込むことができません

マルチスレッド画像の読み込みでは GPU ではなく CPU が使用されるため、pytorch GPU バージョンを再ダウンロードする必要があります。

1. conda listコマンド

(上図のトーチ表示バージョンは1.6.0+cu101、1.6.0と表示されていればCPUバージョンがインストールされていることを意味します)

2. https://pytorch.org/get-started/previous-versions/ にリンクして、対応するバージョンのロードを検索します。

pip install torch==1.9.0+cu102 torchvision==0.10.0+cu102 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html