在 Ubuntu 24.04 上使用 PyCharm 搭建深度学习环境,最推荐且稳健的流程是:利用系统自带的 Conda/Mamba 包管理器创建独立虚拟环境,然后在 PyCharm 中将其识别为解释器。
Ubuntu 24.04 默认使用较新的 Python 版本(3.12)和较新的 GCC/NVIDIA 驱动,这在深度学习环境中可能会遇到兼容性挑战(特别是旧版 CUDA 或某些依赖库)。因此,强烈建议不要直接使用系统自带的 Python,而是通过 Conda 隔离环境。
以下是完整的操作指南:
第一步:安装基础工具
PyCharm 本身不包含 Python 解释器,你需要先确保系统中有 Python 和 Conda。
-
安装 Miniconda (推荐) 或 Anaconda
- 访问 Miniconda 官网 下载 Linux 版本安装包。
- 在终端执行安装(假设文件名为
Miniconda3-latest-Linux-x86_64.sh):bash Miniconda3-latest-Linux-x86_64.sh # 安装过程中按回车同意协议,最后输入 yes 初始化 shell - 重启终端或运行
source ~/.bashrc使环境变量生效。
-
验证安装
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-cpu或tensorflow[and-cuda]。
第三步:在 PyCharm 中配置项目与解释器
现在你已经有了独立的环境,接下来告诉 PyCharm 使用它。
-
打开 PyCharm
- 打开你的深度学习项目文件夹(如果没有项目,点击 "New Project")。
-
添加解释器
- 进入设置菜单:
- 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 保存。
- 进入设置菜单:
-
验证环境
- 在 PyCharm 底部的 Terminal 标签页中,你应该能看到
(dl_env)前缀。 - 尝试运行以下代码验证 GPU 是否可用:
import torch print(torch.__version__) print(torch.cuda.is_available()) print(torch.cuda.get_device_name(0)) - 如果输出
True并显示显卡型号,说明环境搭建成功。
- 在 PyCharm 底部的 Terminal 标签页中,你应该能看到
第四步:处理 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 体验和调试功能。
按照上述步骤,你将获得一个干净、可复现且与系统其他部分解耦的深度学习开发环境。如果遇到具体的报错(如 ModuleNotFoundError 或 CUDA error),请提供错误日志以便进一步排查。
轻量云Cloud