在centos系统中将pytorch升级至最新版,可参考以下操作流程:
卸载已有版本的PyTorch
pip uninstall torch torchvision torchaudio
安装最新的PyTorch版本
前往PyTorch官方网站,挑选适用于你环境的安装指令。
对于CentOS系统,通常可选用如下命令之一(依据CUDA版本而定):
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
若仅需CPU支持,则使用:
pip install torch torchvision torchaudio
注意将cu118更改为你的实际CUDA版本号(如cu113、cu116等)。
检查安装结果
python -c "import torch; print(torch.__version__)"
如果你正在使用Anaconda或Miniconda,可以借助conda来更新PyTorch:
升级conda工具
conda update conda
建立新的conda虚拟环境(非必需)
conda create -n pytorch_env python=3.9 conda activate pytorch_env
安装最新版PyTorch
进入PyTorch官网,选择匹配你系统的安装命令。
CentOS用户一般可用如下命令安装:
conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch

仅需CPU版本时,请使用:
conda install pytorch torchvision torchaudio cpuonly -c pytorch
切记将cudatoolkit=11.8替换为实际使用的CUDA版本。
验证安装情况
python -c "import torch; print(torch.__version__)"
按照上述步骤操作后,你应该可以顺利将PyTorch升级到最新稳定版本。若过程中遇到任何问题,建议查阅PyTorch官方文档或者访问社区论坛获取技术支持。