搭建Linux深度学习环境的第一步是选择合适的发行版。推荐使用Ubuntu 20.04 LTS或22.04 LTS,它们拥有良好的社区支持和丰富的软件包资源,适合初学者与进阶用户。安装时建议选择带有图形界面的版本,便于后续操作。
安装完成后,更新系统软件源并升级所有包。打开终端,执行命令:sudo apt update && sudo apt upgrade -y。这一步确保系统处于最新状态,避免后续依赖冲突。
接下来安装NVIDIA驱动。若使用NVIDIA显卡,需确认GPU型号并安装对应驱动。可通过ubuntu-drivers autoinstall自动安装推荐驱动,再通过nvidia-smi验证驱动是否正常加载。
安装CUDA Toolkit是关键步骤。访问NVIDIA官网下载与显卡驱动兼容的CUDA版本(如11.8),使用deb包方式安装。安装后在~/.bashrc中添加环境变量:export PATH=/usr/local/cuda/bin:$PATH,export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH,然后运行source ~/.bashrc使配置生效。

本图由AI生成,仅供参考
安装cuDNN加速库。从NVIDIA开发者网站注册并下载对应CUDA版本的cuDNN,解压后将文件复制到CUDA安装目录下。验证安装可通过Python脚本导入torch或tensorflow,检查是否能识别GPU。
使用Anaconda或Miniconda管理虚拟环境。安装Miniconda后创建独立环境,例如:conda create -n dl_env python=3.9。激活环境:conda activate dl_env。
在环境中安装主流深度学习框架。推荐使用pip安装PyTorch:pip install torch torchvision torchaudio –index-url https://download.pytorch.org/whl/cu118。或安装TensorFlow:pip install tensorflow[and-cuda]。
•测试环境是否可用。编写一个简单脚本,创建张量并在GPU上执行运算,确认GPU被正确调用。若输出显示GPU信息且计算速度显著提升,则环境搭建成功。