ubuntu安装torch1.7.1

在Ubuntu上安装PyTorch 1.7.1可以分为几个步骤，因为官方版本库可能不包含这么旧的版本，你需要从源码或通过conda来安装。这里提供一种通过conda安装的方式：

1. **更新系统和安装依赖**:

   sudo apt-get update
   sudo apt-get install -y build-essential cmake pkg-config libncurses5-dev libhdf5-dev luarocks libopenblas-dev libcurl4-openssl-dev git

2. **创建一个新的conda环境** (如果你还没有安装Anaconda):

   conda create -n torch1.7 python=3.6
   conda activate torch1.7

3. **安装Miniconda和conda-forge**:

   wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
   bash Miniconda3-latest-Linux-x86_64.sh -b -p $HOME/miniconda
   export PATH="$HOME/miniconda/bin:$PATH"
   conda config --add channels conda-forge

4. **安装PyTorch**:

   conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch -c defaults

这里假设你的CUDA版本是10.1，如果不确定，可以查看`nvidia-smi`命令来确认。

5. **检查安装**:

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

如果需要特定版本1.7.1，由于这个版本可能不再维护，你可能需要下载对应版本的whl文件手动安装，或者寻找社区提供的预编译包。