要在CentOS系统上升级PyTorch版本,您可以按照以下步骤操作:
sudo yum update -y
sudo yum install -y gcc openssl-devel bzip2-devel libffi-devel
sudo yum install -y python3 python3-pip
您可以从Miniconda官网下载适合您系统的安装包,并按照安装向导进行安装。
conda create -n torch_env python3.8
conda activate torch_env
conda install pytorch torchvision torchaudio cudatoolkit=your_cuda_version -c pytorch
请将 your_cuda_version
替换为您的系统上安装的CUDA版本,例如 cudatoolkit11.1
。
pip install torch torchvision torchaudio
python -c "import torch; print(torch.__version__)"
print(torch.cuda.is_available())
如果一切正常,您应该能看到PyTorch的版本号以及CUDA是否可用(取决于您的系统配置)。
请注意,以上步骤可能会因系统版本、CUDA版本和PyTorch版本的不同而有所变化。建议查阅PyTorch的官方文档以获取最准确的安装指南。