ubuntu18.04上有 CUDA 11.4，安装 PyTorch

您可以按照以下步骤在Ubuntu 18.04上安装PyTorch（适用于CUDA 11.4）：

1. 安装CUDA 11.4

首先，您需要在Ubuntu 18.04上安装CUDA 11.4。您可以从NVIDIA官网下载适用于Ubuntu 18.04的CUDA 11.4安装包，并按照官方文档中的安装指南进行安装。

2. 安装cuDNN

安装完成CUDA 11.4后，您需要安装cuDNN。您可以从NVIDIA官网下载适用于CUDA 11.4的cuDNN安装包，并按照官方文档中的安装指南进行安装。

3. 安装PyTorch

在安装完CUDA 11.4和cuDNN后，您可以使用以下命令安装PyTorch：

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

该命令会安装PyTorch、torchvision和torchaudio，并且会使用适用于CUDA 11.0的PyTorch预构建二进制文件。

4. 验证安装

安装完成后，您可以使用以下命令在Python中验证PyTorch是否已正确安装：

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

如果输出显示了PyTorch的版本号，则表示PyTorch已成功安装。

希望这些步骤能够帮助您在Ubuntu 18.04上安装PyTorch。