要在 CentOS 上安裝 PyTorch,您可以使用以下步驟:
sudo yum update
sudo yum install python3 python3-pip
安裝 Conda(如果尚未安裝),這是一個 Python 包管理器,可以簡化安裝和管理 Python 軟件包的過程。要安裝 Miniconda(一個較小的 Anaconda 發行版),請訪問 https://docs.conda.io/en/latest/miniconda.html 并按照說明進行操作。
使用 Conda 創建一個新的 Python 環境(可選,但推薦):
conda create -n torch_env python=3.8
conda activate torch_env
conda install pytorch torchvision torchaudio cpuonly -c pytorch
要安裝支持 GPU 的 PyTorch,請運行:
conda install pytorch torchvision torchaudio cudatoolkit=<your_cuda_version> -c pytorch
請將 <your_cuda_version> 替換為您的系統上安裝的 CUDA 版本,例如 cudatoolkit=11.1。
python -c "import torch; print(torch.__version__)"
這將輸出已安裝的 PyTorch 版本?,F在您可以在 CentOS 上使用 PyTorch 了!