在开始配置前,请确保您的Linux系统满足以下基本要求: - 推荐Ubuntu 18.04/20.04或CentOS 7/8 - 至少8GB RAM(16GB以上更佳) - 支持CUDA的NVIDIA GPU(如需GPU加速) - Python 3.6或更高版本
sudo snap install pycharm-professional --classic
bash
wget https://download.jetbrains.com/python/pycharm-professional-2021.3.2.tar.gz
bash
tar -xzf pycharm-professional-*.tar.gz -C /opt/
cd /opt/pycharm-*/bin
./pycharm.sh
运行PyCharm后,在欢迎界面选择"Tools" > "Create Desktop Entry"
python3 -m venv ~/venv/dl_env
source ~/venv/dl_env/bin/activate
~/venv/dl_env/bin/python
sudo apt update
sudo apt install -y python3-dev python3-pip build-essential cmake git
# CUDA Toolkit (根据您的显卡选择版本)
sudo apt install -y nvidia-cuda-toolkit
# cuDNN (需要从NVIDIA官网下载对应版本)
pip install --upgrade pip
# TensorFlow
pip install tensorflow-gpu # GPU版本
# 或
pip install tensorflow # CPU版本
# PyTorch (访问https://pytorch.org获取最新安装命令)
pip install torch torchvision torchaudio
# 其他常用库
pip install numpy pandas matplotlib scikit-learn jupyter notebook
编辑PyCharm的启动脚本(通常位于/opt/pycharm/bin/pycharm64.vmoptions
):
-Xms1024m
-Xmx4096m
-XX:ReservedCodeCacheSize=1024m
创建测试文件test_dl.py
:
import tensorflow as tf
import torch
# TensorFlow测试
print("TensorFlow version:", tf.__version__)
print("GPU available:", tf.config.list_physical_devices('GPU'))
# PyTorch测试
print("\nPyTorch version:", torch.__version__)
print("CUDA available:", torch.cuda.is_available())
运行该脚本确认所有组件正常工作。
nvidia-smi
tf.config.experimental.set_memory_growth
动态分配GPU内存通过以上步骤,您应该已经成功在Linux系统上配置好了PyCharm用于深度学习开发环境。