確定環境已經安裝 CUDA / cuDNN 之後, 還是無法存取 CUDA:
- CUDA Toolkit: https://developer.nvidia.com/cuda-toolkit
- Installing cuDNN on Windows:
https://docs.nvidia.com/deeplearning/cudnn/latest/installation/windows.html
查看是否有cuda
檢查 cuda 是否可以存取方式如以下：
import torch
if torch.cuda.is_available():
print(True)
else:
print(False)
發現是 pytorch 版本安裝錯誤, 安裝到 CPU 版本:
測試用指令:
torch.__version__顯示為 cpu 版本, 所以無法使用.
pytorch 安裝方式:
https://pytorch.org/
Run this Command:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118從 python 移除 pytorch package 指令:
python -m pip uninstall torch torchvision torchaudio檢視 python 已安裝的 package 指令:
python -m pip list服用 pytorch 網站上的 command 之後, 就可以正常使用 torch 的 GPU了:
