如何快速搭建PyTorch Linux开发环境,轻松入门深度学习,成为深度学习高手?
- 内容介绍
- 文章标签
- 相关推荐
深度学习开发者的痛点:环境搭建太麻烦!
刚接触PyTorch的你是否遇到过这些问题?怎么说呢,
- Linux环境配置复杂,容易出错?
- GPU驱动安装失败,无法加速训练?
- 依赖冲突频繁,项目无法正常运行?不过,
- 不知道怎么选合适的IDE进行高效开发?
别担心!
一、准备工作:确保基础条件齐全
在开始之前,请确保你的Linux程序满足以下基本要求:
- Python 3.8+
- conda/Anaconda/miniconda
- 可选:NVIDIA GPU + CUDA
- 网络连接
1.1 检查Python版本
$ python3 --version
Python 3.8.5 # 示例输出
1.2 安装conda
$ wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
$ bash Miniconda3-latest-Linux-x86_64.sh
$ source ~/.bashrc # 激活conda
$ conda --version # 验证安装成功
conda 4.9.0 # 示例输出
二、创建隔离环境:避免依赖冲突地狱!
"我每次切换项目都要重新设置环境,太烦了!"
别再这样做了!使用虚拟环境可以轻松解决这个问题。这里我们提供两种方案: A) conda虚拟环境 B) Python原生venv
A) conda虚拟环境方案
$ conda create -n pytorch_env python=3.8 # 创建名为pytorch_env的环境
$ conda activate pytorch_env # 激活环境
$ python --version # 验证Python版本匹配
Python 3.8.x # 应与创建时指定版本一致
$ deactivate # 清除时退出环境
$
# 注意终端提示符变化表示激活状态!# 表示已激活该虚拟环境!
话说回来,# 没有提示符则处于base基础环境中!
为什么选择conda?不过,
- ✅ 一键解决多包依赖关系 ✅ 预编译包更快速安装 ✅ 跨网站支持更好 ✅ 自带包管理器功能
-
💡 对于TensorFlow/PyTorch等科学计算包尤其友好!
小心陷阱!
❗ 不要混用conda和pip安装同一个包!❗ 不要直接修改base基础环境!其实,❗ 每次启动终端记得先激活虚拟环境!
B) Python原生venv方案
$ python3 -m venv pytorch_venv # 创建虚拟环境目录pytorch_venv $ source pytorch_venv/bin/activate # 激活 三、PyTorch主要安装: 一键搞定深度学习引擎!"我到底应该安装CPU版还是GPU版PyTorch?" A) 有NVIDIA GPU机器 $ conda install pytorch torchvision torchaudio cudatoolkit=your_cuda_version-c pytorch- CUDA 11.x: cudatoolkit=cudatoolkit==11.x
- CUDA 10.x: cudatoolkit=cudatoolkit==10.x
说到提示,在nvidia-smi命令中查看您当前显卡支持哪个CUDA版本!
验证GPU可用性
$ python -c "import torch;print)" True
B) 唯有CPU可用机器
$ conda install pytorch torchvision torchaudio cpuonly -c pytorch
C) Nightly预览版/
$ conda install -c https://download.pytorch.org/whl/torchnightly/cpu torch torchvision torchaudio
技巧:- 添加-c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pytorch/ - 若报错"no package found",尝试更新conda: $ conda update -n base condalabel='/bgcolor'/
四、GPU驱动检查与调整设置
"我的GPU显卡没被识别怎么办?"
+-----------------------------------------------------------------------------+ | NVIDIA-SMI .... | Driver Version... : ... | CUDA Version..... : ... +---------------------... | Process ID........ : ... | GPU Memory Usage... : ... +-----------------------------------------------------------------------------+
场景/b>/td>(td align="center">可能原因/b>/td>(td align="center">方法/b>/td>( 未检测到任何显卡信息/td>/td>( CUDA版本过低/td>(td align="" width="">部分旧显卡不支持最新CUDA/td>(td align="" width="">降级至支持的最高CUDA版本或升级硬件 ( 权限问题/td>(td align="" width="">普通使用者无权限调用显卡资源/td=(td align="" width=>sudo usermod -aG video $USER && newgrp video /td< (tr bgcolor="#FFFFFF">
深度学习开发者的痛点:环境搭建太麻烦!
刚接触PyTorch的你是否遇到过这些问题?怎么说呢,
- Linux环境配置复杂,容易出错?
- GPU驱动安装失败,无法加速训练?
- 依赖冲突频繁,项目无法正常运行?不过,
- 不知道怎么选合适的IDE进行高效开发?
别担心!
一、准备工作:确保基础条件齐全
在开始之前,请确保你的Linux程序满足以下基本要求:
- Python 3.8+
- conda/Anaconda/miniconda
- 可选:NVIDIA GPU + CUDA
- 网络连接
1.1 检查Python版本
$ python3 --version
Python 3.8.5 # 示例输出
1.2 安装conda
$ wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
$ bash Miniconda3-latest-Linux-x86_64.sh
$ source ~/.bashrc # 激活conda
$ conda --version # 验证安装成功
conda 4.9.0 # 示例输出
二、创建隔离环境:避免依赖冲突地狱!
"我每次切换项目都要重新设置环境,太烦了!"
别再这样做了!使用虚拟环境可以轻松解决这个问题。这里我们提供两种方案: A) conda虚拟环境 B) Python原生venv
A) conda虚拟环境方案
$ conda create -n pytorch_env python=3.8 # 创建名为pytorch_env的环境
$ conda activate pytorch_env # 激活环境
$ python --version # 验证Python版本匹配
Python 3.8.x # 应与创建时指定版本一致
$ deactivate # 清除时退出环境
$
# 注意终端提示符变化表示激活状态!# 表示已激活该虚拟环境!
话说回来,# 没有提示符则处于base基础环境中!
为什么选择conda?不过,
- ✅ 一键解决多包依赖关系 ✅ 预编译包更快速安装 ✅ 跨网站支持更好 ✅ 自带包管理器功能
-
💡 对于TensorFlow/PyTorch等科学计算包尤其友好!
小心陷阱!
❗ 不要混用conda和pip安装同一个包!❗ 不要直接修改base基础环境!其实,❗ 每次启动终端记得先激活虚拟环境!
B) Python原生venv方案
$ python3 -m venv pytorch_venv # 创建虚拟环境目录pytorch_venv $ source pytorch_venv/bin/activate # 激活 三、PyTorch主要安装: 一键搞定深度学习引擎!"我到底应该安装CPU版还是GPU版PyTorch?" A) 有NVIDIA GPU机器 $ conda install pytorch torchvision torchaudio cudatoolkit=your_cuda_version-c pytorch- CUDA 11.x: cudatoolkit=cudatoolkit==11.x
- CUDA 10.x: cudatoolkit=cudatoolkit==10.x
说到提示,在nvidia-smi命令中查看您当前显卡支持哪个CUDA版本!
验证GPU可用性
$ python -c "import torch;print)" True
B) 唯有CPU可用机器
$ conda install pytorch torchvision torchaudio cpuonly -c pytorch
C) Nightly预览版/
$ conda install -c https://download.pytorch.org/whl/torchnightly/cpu torch torchvision torchaudio
技巧:- 添加-c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pytorch/ - 若报错"no package found",尝试更新conda: $ conda update -n base condalabel='/bgcolor'/
四、GPU驱动检查与调整设置
"我的GPU显卡没被识别怎么办?"
+-----------------------------------------------------------------------------+ | NVIDIA-SMI .... | Driver Version... : ... | CUDA Version..... : ... +---------------------... | Process ID........ : ... | GPU Memory Usage... : ... +-----------------------------------------------------------------------------+
场景/b>/td>(td align="center">可能原因/b>/td>(td align="center">方法/b>/td>( 未检测到任何显卡信息/td>/td>( CUDA版本过低/td>(td align="" width="">部分旧显卡不支持最新CUDA/td>(td align="" width="">降级至支持的最高CUDA版本或升级硬件 ( 权限问题/td>(td align="" width="">普通使用者无权限调用显卡资源/td=(td align="" width=>sudo usermod -aG video $USER && newgrp video /td< (tr bgcolor="#FFFFFF">

