Open2022/12/02にコメント追加2

WSL2 + whisper

Windwos 11のWSL2でwhisperを動かそうとして断念した記録

$ pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

ホスト(windows)でGPUドライバ入ってればWSLのubuntu側でCUDA入れなくていいよ、っていうか入れるなって書いてある

import torch

print(torch.__version__)
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.cuda.current_device())
print(torch.cuda.get_device_name())
print(torch.cuda.get_device_name(torch.device('cuda:0')))
print(torch.cuda.get_device_name('cuda:0'))

cnDNNが必要そうだったので入れてみたんだけど、相変わらずエラーが出ちゃう

$ whisper test.mp3 --language Japanese
Could not load library libcudnn_cnn_infer.so.8. Error: libcuda.so: cannot open shared object file: No such file or directory
Please make sure libcudnn_cnn_infer.so.8 is in your library path!
fish: Job 1, 'whisper test.mp3…' terminated by signal SIGABRT (Abort)

ikaruga

WSL2を通さずにWindowsにPythonをインストールするところから初めて動いているのでVMとホストの間でCUDA周りがうまく連携できていなかったのでしょう。