Ubuntu 18.04下CUDA 11.4和PyTorch 11.4的安装教程

在Ubuntu 18.04下安装CUDA 11.4和PyTorch 11.4需要按照一定的步骤进行。首先，确保你的系统已经更新到最新版本。然后，你可以按照以下步骤进行安装：
步骤一：安装CUDA 11.4
安装CUDA 11.4需要先下载相应的安装包。你可以从NVIDIA官网下载适用于Ubuntu的CUDA 11.4安装包。下载完成后，按照以下步骤进行安装：

1. 打开终端，进入安装包所在目录。
2. 解压安装包，运行以下命令：

   sudo apt-get install linux-headers-$(uname -r)
   sudo sh cuda_11.4.0-linux-x64_installers/runfile.sh

3. 安装完成后，你需要设置环境变量。打开终端，输入以下命令：

   echo 'export PATH=/usr/local/cuda-11.4/bin:$PATH' >> ~/.bashrc
   echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
   source ~/.bashrc

   步骤二：安装PyTorch 11.4
   安装PyTorch 11.4需要先安装依赖项。在终端中输入以下命令：

   sudo apt-get install libpython3-dev python3-pip python3-dev

   然后，你可以使用以下命令安装PyTorch 11.4：

   pip3 install torch==1.10.0 torchvision==0.9.0 -f https://download.pytorch.org/whl/cu114/torch_stable.html

   如果你想安装GPU版本的PyTorch，你需要先安装CUDA 11.4，然后使用以下命令：

   pip3 install torch-cuda==1.10.0 torchvision-cuda==0.9.0 -f https://download.pytorch.org/whl/cu114/torch_stable.html

   最后，你可以通过运行以下命令验证安装是否成功：

   import torch
   torch.__version__

   如果输出PyTorch的版本号，说明安装成功。
   以上是在Ubuntu 18.04下安装CUDA 11.4和PyTorch 11.4的简单教程。希望对你有所帮助。