速卖通素材
努力

ubuntu24.04上使用pycharm中搭建深度学习环境?

服务器

在 Ubuntu 24.04 上使用 PyCharm 搭建深度学习环境,最推荐且稳健的流程是:利用系统自带的 Conda/Mamba 包管理器创建独立虚拟环境,然后在 PyCharm 中将其识别为解释器。

Ubuntu 24.04 默认使用较新的 Python 版本(3.12)和较新的 GCC/NVIDIA 驱动,这在深度学习环境中可能会遇到兼容性挑战(特别是旧版 CUDA 或某些依赖库)。因此,强烈建议不要直接使用系统自带的 Python,而是通过 Conda 隔离环境。

以下是完整的操作指南:

第一步:安装基础工具

PyCharm 本身不包含 Python 解释器,你需要先确保系统中有 Python 和 Conda。

  1. 安装 Miniconda (推荐) 或 Anaconda

    • 访问 Miniconda 官网 下载 Linux 版本安装包。
    • 在终端执行安装(假设文件名为 Miniconda3-latest-Linux-x86_64.sh):
      bash Miniconda3-latest-Linux-x86_64.sh
      # 安装过程中按回车同意协议,最后输入 yes 初始化 shell
    • 重启终端或运行 source ~/.bashrc 使环境变量生效。
  2. 验证安装

    conda --version
    python --version

第二步:创建深度学习专用环境

由于深度学习框架(如 PyTorch, TensorFlow)对 CUDA 版本和 Python 版本有特定要求,我们创建一个独立的 Conda 环境。

场景 A:使用 NVIDIA GPU (推荐)
假设你使用的是 NVIDIA 显卡,且已安装对应版本的驱动。我们需要指定 CUDA 版本(例如 CUDA 12.1)。

# 创建名为 'dl_env' 的环境,使用 Python 3.10 (PyTorch 2.x 目前最稳定版本通常是 3.9-3.10)
conda create -n dl_env python=3.10

# 激活环境
conda activate dl_env

# 安装 PyTorch (以官方推荐的 CUDA 12.1 为例)
# 注意:请根据你的显卡驱动支持的最高 CUDA 版本选择,或者让 PyTorch 自动匹配
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

场景 B:仅 CPU 模式 (无显卡或测试用)

conda create -n dl_env_cpu python=3.10
conda activate dl_env_cpu
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

提示:如果需要使用 TensorFlow,可以在上述命令后追加 tensorflow-cputensorflow[and-cuda]

第三步:在 PyCharm 中配置项目与解释器

现在你已经有了独立的环境,接下来告诉 PyCharm 使用它。

  1. 打开 PyCharm

    • 打开你的深度学习项目文件夹(如果没有项目,点击 "New Project")。
  2. 添加解释器

    • 进入设置菜单:
      • File > Settings (Windows/Linux) 或 PyCharm > Settings (macOS)。
      • 导航到 Project: <你的项目名称> > Python Interpreter
    • 点击右上角的齿轮图标 ⚙️,选择 Add…
    • 在左侧列表中选择 Conda Environment -> Existing environment
    • 在右侧的路径框中,填入 Conda 环境的 Python 路径。通常位于:
      /home/你的用户名/miniconda3/envs/dl_env/bin/python
      (如果是 Mamba 安装,路径可能略有不同)
    • 点击 OK 保存。
  3. 验证环境

    • 在 PyCharm 底部的 Terminal 标签页中,你应该能看到 (dl_env) 前缀。
    • 尝试运行以下代码验证 GPU 是否可用:
      import torch
      print(torch.__version__)
      print(torch.cuda.is_available())
      print(torch.cuda.get_device_name(0))
    • 如果输出 True 并显示显卡型号,说明环境搭建成功。

第四步:处理 Ubuntu 24.04 特有的潜在问题

Ubuntu 24.04 引入了较新的系统组件,可能会遇到以下情况:

  • NVIDIA 驱动版本
    确保你安装了最新的专有驱动。在 Ubuntu 24.04 中,可以通过“软件和更新” -> “附加驱动”查看,或者在终端运行:

    ubuntu-drivers devices
    sudo apt install nvidia-driver-550  # 示例版本号,根据实际推荐安装

    重启电脑后,再次检查 nvidia-smi 是否正常。

  • CUDA Toolkit 冲突
    如果使用 Conda 安装 PyTorch,Conda 会自带一个轻量级的 CUDA 运行时库(通常在 libcuda.so),这通常能避免与系统安装的 CUDA Toolkit 冲突。尽量避免在同一个环境中手动 apt install cuda-toolkit,除非你有特殊需求。

  • 权限问题
    如果在 PyCharm 中无法安装 pip 包,可能是权限问题。确保你在 Conda 环境下操作,而不是 root 用户。如果必须安装系统级依赖(如 libgl1-mesa-glx),请在激活环境后使用 sudo apt install ...

总结

在 Ubuntu 24.04 上搭建深度学习环境的核心逻辑是:Conda 负责隔离底层依赖(Python/CUDA),PyCharm 负责提供 IDE 体验和调试功能

按照上述步骤,你将获得一个干净、可复现且与系统其他部分解耦的深度学习开发环境。如果遇到具体的报错(如 ModuleNotFoundErrorCUDA error),请提供错误日志以便进一步排查。

未经允许不得转载:轻量云Cloud » ubuntu24.04上使用pycharm中搭建深度学习环境?