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