Linux系統升級PyTorch到最新版主要有以下兩種方法:
pip install --upgrade pip
。pip install --upgrade torch torchvision torchaudio
,可指定CUDA版本,如pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121
(適用于CUDA 12.1)。conda update conda
。conda update pytorch torchvision torchaudio cudatoolkit=X.X -c pytorch
,將X.X
替換為對應的CUDA版本號,不使用GPU則可省略cudatoolkit
參數。