ubuntu下pytorch怎么使用

在Ubuntu系统上安装并使用PyTorch，你需要遵循几个步骤：

1. **更新包列表**:

   sudo apt-get update

2. **安装依赖库**:
对于GPU支持，首先需要安装CUDA和cuDNN：

   sudo apt-get install nvidia-cuda-toolkit nvidia-cudnn-dev -y

如果没有GPU，可以只安装CPU版本：

   sudo apt-get install python3-pip libncurses5-dev tk-dev build-essential libssl-dev libffi-dev libopenblas-dev libatlas-base-dev gfortran

3. **安装Python和pip**:
Ubuntu已经自带Python，但如果你需要Python 3.x，可以升级到最新版本：

   sudo apt-get install python3 python3-pip

4. **安装PyTorch**:
使用pip安装PyTorch，注意选择与你CUDA版本相对应的版本：

   pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu10X/torch_stable.html

将`cu10X`替换为你实际的CUDA版本。

5. **验证安装**:
进入Python交互模式检查是否安装成功：

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

6. **设置环境变量** (如果有GPU):
需要在终端添加以下行，让系统知道在哪里找到CUDA库：

   export PATH=/usr/local/cuda/bin:$PATH
   export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

完成上述步骤后，你就可以在Ubuntu上使用PyTorch进行深度学习开发了。遇到问题时，记得检查你的CUDA和cuDNN版本是否匹配，并确保所有库都已正确安装。