Debian系統支持PyTorch??赏ㄟ^pip或conda安裝,安裝時需根據系統是否支持GPU選擇對應版本。
安裝步驟:
sudo apt update && sudo apt upgrade
sudo apt install python3 python3-pip
pip install torch torchvision torchaudio
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
(以CUDA 11.7為例)conda create -n pytorch_env python=3.9
conda activate pytorch_env
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
import torch
print(torch.__version__)
print(torch.cuda.is_available()) # 若安裝GPU版本,應返回True
注:安裝前需確保系統已安裝CUDA(若使用GPU)及對應驅動,可參考PyTorch官網指南。