### 安装与配置概述 为了在 Ubuntu 22.04 上成功安装并配置 Python 3.12、CUDA 12.1 和 PyTorch 2.3.0 的兼容环境，以下是详细的说明。 --- #### 1. 更新系统包 确保系统的软件包是最新的，运行以下命令更新系统： ```bash sudo apt update && sudo apt upgrade -y ``` --- #### 2. 安装 Python 3.12 由于默认的 Ubuntu 仓库可能不提供最新的 Python 版本，可以通过编译源码或者使用第三方 PPA 来安装： ##### 使用 DeadSnakes PPA 安装 Python 3.12： ```bash sudo add-apt-repository ppa:deadsnakes/ppa sudo apt update sudo apt install python3.12 python3.12-venv python3.12-dev ``` 验证安装是否成功： ```bash python3.12 --version ``` 创建虚拟环境以便隔离依赖项： ```bash python3.12 -m venv pytorch_env source pytorch_env/bin/activate ``` --- #### 3. 配置 CUDA 12.1 下载 NVIDIA 提供的官方 CUDA 工具包，并按照以下步骤操作： ##### 下载 CUDA 安装程序： ```bash wget https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run chmod +x cuda_12.1.1_530.30.02_linux.run sudo ./cuda_12.1.1_530.30.02_linux.run ``` 注意，在安装过程中可以选择仅安装工具链而不覆盖现有驱动[^3]。 完成安装后，修改 `.bashrc` 文件以添加路径支持： ```bash echo 'export PATH=/usr/local/cuda-12.1/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc source ~/.bashrc ``` 测试 CUDA 是否正常工作： ```bash nvcc --version ``` --- #### 4. 安装 PyTorch 2.3.0 并指定 CUDA 12.1 支持 访问 [PyTorch官网](https://pytorch.org/get-started/previous-versions/) 获取旧版安装指南。对于 CUDA 12.1 和 PyTorch 2.3.0，可以执行以下 pip 命令来安装所需的库版本： ```bash pip install torch==2.3.0 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 ``` 如果需要升级 `matplotlib` 或其他科学计算库，则可参考以下命令： ```bash pip install --upgrade matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple ``` --- #### 5. 测试 GPU 加速功能 编写简单的脚本来确认 PyTorch 正常利用了 GPU 计算资源： ```python import torch print(f"CUDA Available: {torch.cuda.is_available()}") print(f"Current Device: {torch.cuda.current_device()}") print(f"Device Name: {torch.cuda.get_device_name(0)}") if torch.cuda.is_available(): tensor = torch.rand((3, 3)).to('cuda') print(tensor) else: print("GPU is not available.") ``` --- #### 注意事项 - 如果硬件显卡无法支持 CUDA 12.1，则需降级到较低版本（如 CUDA 11.x），这取决于实际设备驱动的支持情况[^4]。 - 推荐优先采用 pip 方式安装 PyTorch，而非 conda 渠道，因后者可能存在兼容性问题[^2]。 --- ###