PyTorch 1.8安装

您好，PyTorch 1.8 是一个深度学习框架，支持在 GPU 和 CPU 上高效地进行张量计算。下面是 PyTorch 1.8 的安装方法：

1. 安装 Anaconda：在 Anaconda 的官网上下载对应系统的安装包，并按照提示安装。

2. 创建虚拟环境：打开 Anaconda Prompt，输入以下命令创建一个名为 pytorch_env 的虚拟环境：

   conda create --name pytorch_env

3. 激活虚拟环境：输入以下命令激活 pytorch_env 环境：

   conda activate pytorch_env

4. 安装 PyTorch：输入以下命令安装 PyTorch：

   conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

如果您没有 NVIDIA 的显卡，则可以将 cudatoolkit=11.1 改为 cpuonly。

5. 验证 PyTorch 安装是否成功：在 Anaconda Prompt 中输入以下命令，如果没有报错则说明安装成功：

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

6. 关闭虚拟环境：输入以下命令关闭虚拟环境：

   conda deactivate

希望这些步骤能够帮助到您。