pytorch1.7.1如何安装

要安装PyTorch 1.7.1，您可以按照以下步骤进行操作：
1. 首先，确保您的CUDA版本与PyTorch 1.7.1兼容。您可以在PyTorch官网上的“Get Started”页面中查看支持的CUDA版本。
2. 打开Anaconda Prompt并进入您的环境。您可以使用以下命令激活您的环境：

conda activate <your_env_name>

3. 运行以下命令来安装PyTorch 1.7.1和相应的CUDA工具包：

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=<your_cuda_version>

请注意，您需要将`<your_cuda_version>`替换为您的CUDA版本号，例如`11.0`。
4. 安装完成后，您可以使用以下命令验证PyTorch是否正确安装：

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

如果输出为`1.7.1`，则表示PyTorch已成功安装。