pytorch cuda安装

要在PyTorch中使用CUDA，需要确保系统中已安装了NVIDIA GPU和CUDA驱动程序。以下是在Linux系统上安装PyTorch和CUDA的基本步骤：

1. 安装NVIDIA驱动程序
在NVIDIA官网上下载适合您GPU和Linux系统版本的最新NVIDIA驱动程序。然后，在终端中运行以下命令以停止X服务器并安装驱动程序：

   sudo systemctl stop lightdm.service
   sudo sh NVIDIA-Linux-x86_64-XXX.XX.run

其中`NVIDIA-Linux-x86_64-XXX.XX.run`是您下载的驱动程序文件名。

2. 安装CUDA Toolkit
在NVIDIA官网上下载适合您GPU和Linux系统版本的最新CUDA Toolkit。然后，在终端中运行以下命令以安装CUDA Toolkit：

   sudo sh cuda_XXX.XX.X_linux.run

其中`cuda_XXX.XX.X_linux.run`是您下载的CUDA Toolkit文件名。

3. 安装PyTorch
在PyTorch官网上选择适合您的CUDA版本和Python版本的PyTorch版本，并按照说明进行安装。例如，在Python 3.7和CUDA 10.1下安装PyTorch 1.7.1：

   pip3 install torch==1.7.1+cu101 -f https://download.pytorch.org/whl/cu101/torch_stable.html

安装完成后，您可以使用以下代码检查是否成功安装了PyTorch和CUDA：

import torch
print(torch.cuda.is_available())

如果输出`True`，则说明已成功安装PyTorch和CUDA。