要在CentOS上更新PyTorch到最新版本,请按照以下步骤操作:
python --version
pip --version
如果没有安装Python和pip,请先安装它们。
sudo yum install epel-release
sudo pip install --upgrade pip
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu111
对于GPU版本(请确保已安装NVIDIA CUDA工具包),运行以下命令:
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu111
请注意,您需要将cu111
替换为您的CUDA版本。您可以在PyTorch官方网站上找到支持的CUDA版本。
python -c "import torch; print(torch.__version__)"
现在,您已经成功地将PyTorch更新到了最新版本。