「CUDAをインストールしていないけど、PyTorchでGPUを使いたい」
一昔前なら、Linux（Ubuntu）の黒画に10回くらい泣いてからようやく実現できたその行為も、今やpipやuvでPyTorchをインストールするだけでCUDA機能が使えるようになっています。簡単に紹介します。
環境
Ubuntu 22.04
NVIDIA GeForce RTX 3060
NVIDIAドライバ: 535.183.01
Python 3.10.12
CUDA: 未インストール
NVIDIAドライバは必要です。LinuxとNVIDIAドライバの具体的なセットアップ方法は、以下記事参照くだ…
Source link
Views: 0
