CUDA 问题。 我认为 GPU 可能检测不到

Question

我尝试安装这个： https://github.com/bycloudai/CVPR2022-DaGAN-Window

并使用

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
to install torch.

有一点问题。

当我尝试在 ANACONDA (conda) 中运行下一个命令时出现的错误：

python demo.py  --config config/vox-adv-256.yaml --driving_video driving/driving.mp4 --source_image input/input.jpg --checkpoint checkpoints/SPADE_DaGAN_vox_adv_256.pth.tar --relative --adapt_scale --kp_num 15 --generator SPADEDepthAwareGenerator

错误：

UserWarning: Arguments other than a weight enum or `None` for 'weights' are deprecated since 0.13 and may be removed in the future. The current behavior is equivalent to passing `weights=None`.
warnings.warn(msg)
Traceback (most recent call last):
File "demo.py", line 165, in <module>
loaded_dict_enc = torch.load('depth/models/weights_19/encoder.pth')
File "C:\Users\Qwepy\anaconda32\envs\DaGAN\lib\site-packages\torch\serialization.py", line 789, in load
return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
File "C:\Users\Qwepy\anaconda32\envs\DaGAN\lib\site-packages\torch\serialization.py", line 1131, in _load
result = unpickler.load()
File "C:\Users\Qwepy\anaconda32\envs\DaGAN\lib\site-packages\torch\serialization.py", line 1101, in persistent_load
load_tensor(dtype, nbytes, key, _maybe_decode_ascii(location))
File "C:\Users\Qwepy\anaconda32\envs\DaGAN\lib\site-packages\torch\serialization.py", line 1083, in load_tensor
wrap_storage=restore_location(storage, location),
File "C:\Users\Qwepy\anaconda32\envs\DaGAN\lib\site-packages\torch\serialization.py", line 215, in default_restore_location
result = fn(storage, location)
File "C:\Users\Qwepy\anaconda32\envs\DaGAN\lib\site-packages\torch\serialization.py", line 182, in _cuda_deserialize
device = validate_cuda_device(location)
File "C:\Users\Qwepy\anaconda32\envs\DaGAN\lib\site-packages\torch\serialization.py", line 166, in validate_cuda_device
raise RuntimeError('Attempting to deserialize object on a CUDA '
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.

我应该怎么办？ 我有带有 RTX 2060 SETUP 的 CUDA 工具包 11.3.1 版本。 我认为这个程序的设置有问题，或者因为它没有检测到我的 GPU。

我尝试重装cuda，和我用pytorch安装的版本匹配，我也重装了pytorch，确定不是cpuonly的。

Answer 1

除了在 conda 环境中与 PyTorch 一起安装的内容之外，您不需要单独安装 CUDA。 请检查 NVIDIA 的 gpu 驱动程序版本以及它们是否与 CUDA 工具包兼容。 如果没有，您必须相应地升级或降级它。

Answer 2

尝试按照此https://pytorch.org/get-started/locally/ 安装最新版本的 pytorch