ubuntu 怎么安装pytorch

要在Ubuntu上安装PyTorch，可以按照以下步骤进行操作：

1. 首先，确保你的系统已经安装了Python和pip。可以通过在终端运行以下命令来检查它们是否已安装：

   python3 --version
   pip3 --version

2. 接下来，使用以下命令安装PyTorch的依赖项：

   sudo apt-get update
   sudo apt-get install libopenblas-dev libblas-dev m4 cmake cython python3-dev python3-yaml python3-setuptools

3. 选择适合你系统的安装命令。如果你的系统支持CUDA（NVIDIA显卡），可以使用以下命令安装PyTorch（如果没有CUDA，可以跳过这一步）：

   pip3 install torch torchvision torchaudio

如果你的系统不支持CUDA，可以使用以下命令安装PyTorch：

   pip3 install torch==1.8.0+cpu torchvision==0.9.0+cpu torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

4. 安装完成后，可以使用以下命令验证PyTorch是否成功安装：

   python3 -c "import torch; print(torch.__version__)"

这样，你就可以在Ubuntu上成功安装PyTorch了。希望对你有帮助！