ubnutu安装pytorch

在Ubuntu系统上安装PyTorch，通常需要先确保已经安装了必要的依赖库，如CUDA（如果需要GPU加速）、cuDNN和Python。以下是详细的步骤：

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

   sudo apt-get update

2. **安装Python和pip**:
如果Python版本过旧，可以升级到最新版：

   sudo apt-get install python3-m venv torch_env
   source torch_env/bin/activate

3. **安装必要的库**:

   sudo apt-get install build-essential cmake pkg-config libopenblas-dev luarocks libncurses5-dev libncursesw5-dev libreadline6-dev libgdbm-dev libsqlite3-dev libssl-dev libbz2-dev libexpat1-dev liblzma-dev zlib1g-dev libffi-dev

4. **为CUDA和cuDNN做准备**:
首先检查是否有CUDA支持：

   nvidia-smi

如果有CUDA，访问NVIDIA官网下载对应版本的cuDNN，并将cuDNN文件夹放入`/usr/local/cuda/include`和`/usr/local/cuda/lib64`。

5. **安装PyTorch**:
使用pip安装预编译的CPU版本或CUDA版本，例如：
- CPU版本（无GPU）:

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

- GPU版本（如果有CUDA）:

     pip install torch torchvision torchaudio --extra-index-url=https://download.pytorch.org/whl/cu102/torch_stable.html

6. **确认安装**:
安装完成后，可以运行以下代码来测试安装是否成功：

   import torch
   print(torch.__version__)

记得根据你的硬件和具体需求调整上述步骤。如果你遇到问题，可以在安装过程中查阅PyTorch官方文档或社区论坛寻求帮助。