python3.6安装pytorch

在 Python 3.6 中安装 PyTorch 有两种方法。

方法一：使用 pip 安装

在命令行中运行以下命令：

pip3 install torch==1.8.0+cpu torchvision==0.9.0+cpu torchaudio===0.8.0 -f https://download.pytorch.org/whl/cu101/torch_stable.html

如果你的机器支持 GPU 加速，可以将 `cpu` 替换为 `cu101`，下载相应的 CUDA 版本。

方法二：从源代码构建

首先，你需要安装 PyTorch 的依赖项。在命令行中运行以下命令：

sudo apt-get update
sudo apt-get install libopenblas-dev libblas-dev m4 cmake cython python3-dev python3-yaml python3-setuptools

然后，从 PyTorch 的 GitHub 上下载源代码并解压：

wget https://github.com/pytorch/pytorch/archive/v1.8.0.tar.gz
tar xvzf v1.8.0.tar.gz

进入解压后的目录，运行以下命令：

export NO_CUDA=1
python3 setup.py install

如果你的机器支持 GPU 加速，需要将 `NO_CUDA` 设为 `0`。

注意：从源代码构建 PyTorch 可能需要一些时间，因为需要编译 C++ 代码。同时，需要确保你的机器上已经安装了 C++ 编译器和其他必要的工具。