pytorch-cuda安装

PyTorch是一个非常流行的深度学习框架，而CUDA是NVIDIA提供的用于并行计算的平台和API。使用PyTorch进行深度学习时，借助CUDA可以加速模型的训练和推理。

下面是在Ubuntu系统下安装PyTorch和CUDA的步骤：

1. 首先，需要安装NVIDIA显卡驱动。可以通过命令行查看当前系统中显卡的型号：

   lspci | grep -i nvidia

然后访问NVIDIA官网，下载对应的驱动程序。

2. 安装CUDA。可以通过NVIDIA官网下载适用于自己的操作系统和显卡型号的CUDA版本，并根据官网提供的安装说明进行安装。

3. 安装cuDNN。cuDNN是一个GPU加速库，可以优化深度学习框架的计算速度。同样可以在NVIDIA官网下载适用于自己CUDA版本的cuDNN库，并根据安装说明进行安装。

4. 安装PyTorch。可以通过官网提供的命令行安装方式，选择适用于自己的系统和CUDA版本的PyTorch版本进行安装：

   pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

如果上述步骤完成后，发现PyTorch无法使用GPU加速，则需要检查是否正确安装了NVIDIA显卡驱动、CUDA和cuDNN，并且将PyTorch中的device设置为cuda。

相关问题：
1. PyTorch和TensorFlow有什么区别？
2. 什么是CUDA？
3. 什么是cuDNN？
4. 如何查看当前系统中显卡的型号？