こんにちは、人です。
今回は【ゼロからの物体検出】第5回です。
CUDAをインストールしていきます。
CUDAとはなんぞやっていう話なんですけど、GPUを使えるようにするためのものっぽいですね。
ちなみにNVIDIAっていう会社から出されているGPUじゃないとダメらしいです。
人のパソコンは運良くNVIDIA製のGPUを積んでたんで使ってみることにしました。
早速インストールしていきましょう。こちらのサイトにアクセスしてください。
こんな感じの画面になりましたでしょうか？↓↓↓
今回はCUDA11.1.1を選びます。
次は環境の選択です。↓↓↓
人はこんな感じですね。
さぁダウンロードしていきましょう。
あとは「agree」を押していくだけです。
（余談ですが、この時にVisual Studioが入ってないとエラーが起きるっぽいです）
うまくいけばこんな感じ↓↓↓
終わりました！
次はPyTorchをインストールしていきましょう！
コメント