PyTorch 的 CUDA 版本怎麼選:Windows 上 torch.cuda.is_available() 為 False 時改裝 CUDA 版
PyTorch 搭配 NVIDIA 顯示卡時,能把 AI 模型的運算交給 GPU 處理,但在 Windows 上用 pip 直接安裝時,裝到的是只支援 CPU 的版本,程式也就偵測不到顯示卡。只要先確認驅動程式支援的 CUDA 版本,再從 PyTorch 官方的下載站重新安裝對應的 CUDA 版,PyTorch 就能用上顯示卡,且不需要另外安裝 CUDA T …