関連するエラー
AttributeError: module 'torch._C' has no attribute '_cuda_setDevice'環境
- Windows 10
- python 3.6
- その他CUDA, cudnn, pytorchなど
- GPU NVIDIA GeForce GTX1080 Ti
解決法
以下のエラーが発生。
Torch not compiled with CUDA enabled調べると、pytorchのGitHubに有益なコメントを発見。
AssertionError: Torch not compiled with CUDA enabled · Issue #30664 · pytorch/pytorch
I'm trying to do neural style swapping, and for some reason, I keep getting the following errors. AssertionError: Torch ...
github.com
このコマンドを実行するとGPUを認識するようになった。
pip install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html