0Python+fastapi，第一个Python项目走向第18步：ubuntu24.04 安装cuda和pytorch环境

在Ubuntu 24.04上安装CUDA和PyTorch的步骤如下：

1. 更新系统包索引：

sudo apt update
sudo apt upgrade

2. 安装CUDA（假设你的显卡支持CUDA）：

   首先，访问NVIDIA官网查看你的显卡支持的CUDA版本，然后根据你的显卡型号安装相应版本的CUDA。

例如，如果你的显卡支持CUDA 11.2，你可以使用以下命令安装CUDA 11.2：

sudo apt install cuda-11-2

3. 设置环境变量：

   在你的.bashrc或.profile文件中添加以下行：

export PATH=/usr/local/cuda-11.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

然后执行：

source ~/.bashrc

4. 安装PyTorch：

   访问PyTorch官网的安装指南（https://pytorch.org/get-started/locally/），选择合适的配置，然后复制给出的安装命令。

例如，如果你的系统配置为CUDA 11.2，你可以使用以下命令安装PyTorch：

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

请确保你的网络连接良好，以便能够顺利下载安装包。

5. 验证安装：

   在Python中验证CUDA是否正确安装：

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

如果输出为True，则表示CUDA已成功安装并且可以通过PyTorch使用。