pytorch1.9.0gpu安装教程

以下是PyTorch 1.9. GPU版的安装教程：

1. 安装CUDA和cuDNN

在安装PyTorch之前，需要先安装CUDA和cuDNN。可以从NVIDIA官网下载对应版本的CUDA和cuDNN，并按照官方文档进行安装。

2. 创建虚拟环境

为了避免与其他Python环境产生冲突，建议在安装PyTorch之前创建一个虚拟环境。可以使用conda或者virtualenv等工具创建虚拟环境。

3. 安装PyTorch

可以使用conda或者pip安装PyTorch。如果使用conda，可以使用以下命令：

conda install pytorch torchvision torchaudio cudatoolkit=<version>

其中，`<version>`是CUDA的版本号，例如`11.1`。如果使用pip，可以使用以下命令：

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu<version>/torch_stable.html

其中，`<version>`同样是CUDA的版本号。

4. 验证安装

安装完成后，可以使用以下代码验证PyTorch是否成功安装：

import torch

print(torch.__version__)
print(torch.cuda.is_available())

如果输出的版本号为1.9.，并且CUDA可用，则说明PyTorch安装成功。

希望这个教程能够帮助到你。如果有任何问题，请随时提问。