Debian下PyTorch支持GPU加速,需安裝NVIDIA顯卡驅動、CUDA Toolkit和cuDNN庫,具體如下:
apt
安裝NVIDIA驅動,如sudo apt install nvidia-driver-<version>
,需根據顯卡型號選擇版本。.deb
文件),通過dpkg
安裝,需配置倉庫密鑰。/usr/local/cuda
)。pip
指定CUDA版本安裝,如pip install torch --extra-index-url https://download.pytorch.org/whl/cu117
(對應CUDA 11.7)。import torch; print(torch.cuda.is_available())
,返回True
表示GPU支持正常。注:需確保CUDA、cuDNN與PyTorch版本兼容,可參考PyTorch官網的版本對照表。