在CentOS上安裝PyTorch的兼容版本,你需要先確保系統已經安裝了Python和pip
sudo yum update -y
sudo yum install -y gcc openssl-devel bzip2-devel libffi-devel wget
sudo yum install -y python3 python3-pip
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu113
這里,cu113
表示你正在使用CUDA 11.3版本。如果你使用的是其他版本的CUDA,請將cu113
替換為相應的版本號。
python3 -c "import torch; print(torch.__version__)"
這將輸出已安裝的PyTorch版本?,F在你可以在CentOS上運行和測試PyTorch了。