以下是在CentOS上配置PyTorch環境的步驟:
sudo yum update -y。sudo yum install -y python3 python3-pip python3-devel,若用CUDA還需安裝相關依賴。python3 -m venv pytorch_env,然后source pytorch_env/bin/activate。pip3 install torch torchvision torchaudio。pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117。import torch,然后print(torch.__version__)和print(torch.cuda.is_available())。