ChatGPT加持，Ubuntu深度学习环境配置指南

ChatGPT加持，Ubuntu深度学习环境配置指南

随着ChatGPT的火爆，越来越多开发者希望在自己的Ubuntu系统上搭建深度学习环境。本文将为您详细介绍如何在Ubuntu 22.04上配置完整的深度学习开发环境，包括NVIDIA显卡驱动、CUDA、Pytorch、TensorFlow，以及常用的IDE如VSCode和PyCharm。无论是初学者还是资深开发者，都能从中受益，快速上手最新技术。

首先确保您的系统已经安装了Ubuntu 22.04。如果尚未安装，可以从Ubuntu官网下载ISO镜像，并使用如Rufus等工具制作启动U盘进行安装。

在安装任何与GPU相关的软件之前，确保您的NVIDIA GPU驱动是最新的。

1. 检查显卡型号：

   sudo lshw -C display
   

2. 更新系统：

   sudo apt update
   sudo apt upgrade
   

3. 添加NVIDIA驱动PPA：

   sudo add-apt-repository ppa:graphics-drivers/ppa
   sudo apt update
   

4. 查找并安装推荐的驱动版本：

   ubuntu-drivers devices
   

   从输出中找到适用于您的显卡的推荐驱动版本，并使用以下命令安装（以nvidia-driver-535为例）：

   sudo apt install nvidia-driver-535
   

5. 重启系统：

   sudo reboot
   

6. 验证驱动安装：

   nvidia-smi
   

   如果命令成功显示NVIDIA显卡和驱动版本的信息，则表示驱动已正确安装。

深度学习框架如TensorFlow和Pytorch需要CUDA和cuDNN的支持才能利用GPU加速。这里我们使用Conda来管理这些依赖。

1. 安装Miniconda：
   访问Miniconda官网下载适用于Linux的Miniconda安装包。下载后在终端中运行安装包：

   bash Miniconda3-latest-Linux-x86_64.sh
   

   按照提示操作，记得将conda添加到您的PATH环境变量中。

安装Pytorch

1. 创建虚拟环境：

   conda create -n pytorch_env python=3.12
   conda activate pytorch_env
   

2. 安装Pytorch：

   pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
   

安装TensorFlow

1. 创建新环境：

   conda create -n tensorflow_env python=3.8
   conda activate tensorflow_env
   

2. 安装CUDA和cuDNN：

   conda install cudatoolkit=11.2 cudnn=8.1.0 -c pytorch
   

3. 安装TensorFlow：

   pip install tensorflow-gpu
   

VSCode配置

1. 下载VSCode：
   访问VSCode官网下载适用于Linux的.deb文件。

2. 安装VSCode：
   双击下载的.deb文件直接安装。如果直接安装失败，可以尝试以下命令：

   sudo dpkg -i code_1.85.1-1699778000_amd64.deb
   sudo apt install -f
   sudo dpkg -i code_1.85.1-1699778000_amd64.deb
   

3. 配置Python解释器：
   打开VSCode，在菜单中选择File > Settings > Python Interpreter，选择之前创建的conda环境。

PyCharm配置

1. 下载PyCharm：
   访问PyCharm官网下载适用于Linux的版本。

2. 解压并运行：
   下载后解压，通过终端进入解压目录，运行./pycharm.sh启动PyCharm。

3. 配置Python解释器：
   在File > Settings > Project: your_project_name > Python Interpreter中，选择Add，然后选择Existing environment，指向之前创建的conda环境。

1. 安装必要的Python包：

   sudo apt update -y
   sudo apt install python3 python3-pip python3-venv -y
   

2. 创建虚拟环境：

   python3 -m venv chatgpt_env
   source chatgpt_env/bin/activate
   

3. 安装ShellGPT：

   pip install shellgpt
   

4. 配置API密钥：

   shellgpt config --api-key YOUR_API_KEY
   

通过以上步骤，您已经成功在Ubuntu 22.04上搭建了一个完整的深度学习开发环境，包括最新的显卡驱动、CUDA支持、深度学习框架以及开发工具。现在，您可以开始您的深度学习之旅了！