Python 使用 GPU 的安裝方式

依據顯卡的資訊

Nvidia-smi 顯示卡 資訊

我的經驗是先更新顯卡的驅動到可支援的最新版本

再到nvida網站下載適合你版本的cuda

CUDA Toolkit Archive | NVIDIA Developer

我是下載12.4版本來做測試

查看是否安裝完成 nvcc -V

在python環境上驗證:

import torch

print(torch.cuda.is_available())