Linux 安装PyTorch

在Linux上安装PyTorch通常需要几个步骤。以下是简要的指南：

1. 更新系统：

   sudo apt-get update
   sudo apt-get upgrade

2. 安装必要的依赖：
对于CUDA支持，你需要安装对应的库和工具包，例如CUDA、cuDNN和gcc。如果你不确定版本，请先确认你的CUDA版本，然后从官方GitHub仓库下载对应版本的PyTorch。

   sudo apt-get install nvidia-cuda nvidia-cudnn libncurses5-dev build-essential cmake python3-dev python3-pip swig

3. 下载并配置PyTorch源码：

   git clone https://github.com/pytorch/pytorch.git
   cd pytorch

4. 配置安装选项，根据需要选择CPU或GPU安装，并指定Python路径（例如，对于CPU安装）：

   export TORCH_CUDA_ARCH_LIST="6.0"
   ./configure --prefix=$HOME/.local/packages/torch --cmake-force-advanced -DCMAKE_BUILD_TYPE=Release -DPYTHON_EXECUTABLE=/usr/bin/python3 -DUSE_NCCL=OFF

5. 编译并安装：

   make -j$(nproc)
   sudo make install

6. 验证安装：

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

如果遇到问题，可能需要查看PyTorch的官方文档或社区资源，因为某些发行版的依赖可能有所不同。