在Ubuntu上更新PyTorch版本,常用方法有以下两种:
pip show torch
查看当前版本。pip install --upgrade pip
。pip uninstall torch torchvision torchaudio
。pip install torch torchvision torchaudio
。若需特定版本,可指定版本号,如pip install torch==<version> torchvision==<version> torchaudio==<version>
。pip show torch
查看版本信息。conda activate <your_env>
。conda list torch
。conda remove pytorch torchvision torchaudio
。conda update pytorch torchvision torchaudio cudatoolkit=<your_cuda_version> -c pytorch
,其中<your_cuda_version>
替换为实际CUDA版本,若不用GPU可省略相关参数。conda list torch
查看版本。