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