アプリケーション開発ポータルサイト
ServerNote.NET
カテゴリー【PythonWindows
Windows版PyTorchでCUDA GPUを使えるようにする(torch.cuda.is_available()=Trueにする)
POSTED BY
2023-05-18

WindowsにPythonとそのモジュールPyTorch、およびNVIDIAのCUDAツールキットを入れて、いざGPUを使おうとすると、

import torch
torch.cuda.is_available();
False

あっさり否定されてGPUが使えない

これは、Windowsのpipでノーオプションで普通に

pip install torch torchvision torchaudio

と、標準のものをインストールしてしまっていることが原因。オプションをつけて、専用サイトのものをインストールする必要がある。

まず既存のそれらを削除する。

pip uninstall torch torchvision torchaudio

そして、以下のように、インストールしているCUDAのバージョンに合ったURLを確認する。

https://download.pytorch.org/whl/cu118

ブラウザに上記を貼り付けて、ページが開けてモジュール一覧が表示されていればそのCUDAバージョン用のtorchが存在するということになる。当方11.8を入れているので、上記cu118となる。

確認できたらあらためてpipにオプションをつけてインストールする。

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118

これで、晴れてPyTorchからCUDA-GPUが使用できるはずだ。

import torch
torch.cuda.is_available();
True

※本記事は当サイト管理人の個人的な備忘録です。本記事の参照又は付随ソースコード利用後にいかなる損害が発生しても当サイト及び管理人は一切責任を負いません。
※本記事内容の無断転載を禁じます。
【WEBMASTER/管理人】
自営業プログラマーです。お仕事ください!
ご連絡は以下アドレスまでお願いします★

【キーワード検索】