「AssertionError: Torch not compiled with CUDA enabled」とpytorchがGPUを認識しない時

Pytorchでcudaが使えない時どうするか？

　Pytorchをconda環境で使っているときに、

AssertionError: Torch not compiled with CUDA enabled

エラーに出会うことがあります。
そんな時は、

conda install -c pytorch pytorch

とすると解決します。

詳細な原因

　詳細な原因と解決策は以下の通りです。

エラーの詳細

エラー文

>>> import torch
>>> torch.cuda.current_device()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/hogehoge/yes/envs/gpu_test/lib/python3.7/site-packages/torch/cuda/__init__.py", line 479, in current_device
    _lazy_init()
  File "/home/hogehoge/yes/envs/gpu_test/lib/python3.7/site-packages/torch/cuda/__init__.py", line 208, in _lazy_init
    raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

原因

　エラーの原因はPytorchはcudaサポートがあるバージョンとないバージョンがあるためにGPUが認識しないと言うのが原因らしいです。

解決策

　解決策は、以下のコマンドを、好きなconda環境で　打つことらしいです。
なんで解決するかは知りません。

 conda install -c pytorch pytorch

参考

AssertionError: Torch not compiled with CUDA enabled #30664