PyTorch在Ubuntu上支持CUDA吗

是的，PyTorch在Ubuntu上支持CUDA。要在Ubuntu上使用PyTorch的GPU加速功能，需要安装CUDA Toolkit和cuDNN库，并确保PyTorch版本与CUDA版本兼容。以下是相关信息：

安装步骤

1. 安装NVIDIA显卡驱动：首先需要安装与您的NVIDIA显卡型号相匹配的驱动程序。
2. 安装CUDA Toolkit：根据您的PyTorch版本选择合适的CUDA Toolkit版本进行安装。可以从NVIDIA官方网站下载并安装。
3. 安装cuDNN：下载与CUDA Toolkit版本兼容的cuDNN库，并按照官方指南进行安装。
4. 安装PyTorch：使用conda或pip安装PyTorch，并指定CUDA版本。例如，要为CUDA 12.1安装PyTorch，可以使用以下命令：

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

或者使用pip：

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu121/torch_stable.html
```。

### 验证安装

安装完成后，可以通过运行以下Python代码来验证PyTorch是否能够使用GPU：

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

如果输出为True，则表示PyTorch已成功配置为使用GPU。

请注意，为了获得最佳的性能和兼容性，建议在安装PyTorch和CUDA之前，查阅PyTorch的官方文档，了解当前版本支持的CUDA版本和具体的安装指南。