ubuntu24.04安装pytorch

在Ubuntu 24.04上安装PyTorch，你需要先确保你的系统已经更新到最新版本，并且安装了必要的依赖。以下是安装步骤：

1. **安装Python基础库**：

   sudo apt-get update
   sudo apt-get install python3-pip

2. **安装CUDA和cuDNN** (如果你需要GPU支持)：
- 首先检查你的显卡是否兼容CUDA：

     lspci | grep -i nvidia

- 官方CUDA下载地址：https://developer.nvidia.com/cuda-downloads
- 按照GPU型号选择合适的cuDNN版本：https://developer.nvidia.com/cudnn
- 安装CUDA和cuDNN，然后配置环境变量。

3. **安装PyTorch**：
使用pip安装CPU版本的PyTorch：

   pip3 install torch torchvision torchaudio cpuonly -f https://download.pytorch.org/whl/lts/nightly/torch_lts.html

或者安装支持GPU的PyTorch（如果有安装CUDA和cuDNN）：

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

4. **验证安装**：

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