在Ubuntu上更新PyTorch庫,你可以使用pip
或conda
(如果你使用的是Anaconda發行版)。以下是兩種方法的步驟:
打開終端(Terminal)。
首先,確保你的pip版本是最新的。你可以使用以下命令來更新pip:
pip install --upgrade pip
然后,你可以使用以下命令來更新PyTorch:
pip install --upgrade torch torchvision torchaudio
這將安裝最新版本的PyTorch及其相關庫。
如果你使用的是Anaconda,可以使用以下步驟來更新PyTorch:
打開終端(Terminal)。
更新conda到最新版本:
conda update conda
更新PyTorch庫:
conda update pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
注意:cudatoolkit=11.3
是CUDA工具包的版本,你需要根據你的GPU和CUDA版本來選擇合適的工具包版本。如果你不需要GPU支持,可以省略cudatoolkit
部分。
在執行這些命令之前,請確保你的系統已經安裝了Python和相應的包管理器(pip或conda)。如果你還沒有安裝它們,可以通過以下命令安裝:
對于pip:
sudo apt update
sudo apt install python3-pip
對于conda:
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
按照提示完成安裝過程。
更新完成后,你可以使用pip list
或conda list
命令來檢查PyTorch庫的新版本是否已經安裝成功。