Linux Pytorch深度学习环境搭建速通 • Pdch's Blog

目录#

Step 0. 系统启动盘制作
Step 1. 终端美化与代理设置
Step 2. 安装NVIDIA驱动
Step 3. 安装python环境管理工具

Step 0. 系统启动盘制作#

简单来说，Ventoy是一个制作可启动U盘的开源工具。有了Ventoy你就无需反复地格式化U盘，你只需要把 ISO/WIM/IMG/VHD(x)/EFI 等类型的文件直接拷贝到U盘里面就可以启动了，无需其他操作。你可以一次性拷贝很多个不同类型的镜像文件，Ventoy 会在启动时显示一个菜单来供你进行选择。下载地址:https://www.ventoy.net/cn/download.html ↗ 安装到U盘以后，只需要将iso镜像复制到U盘中即可。个人建议前往镜像站下载操作系统的iso镜像。

安装过程因人而异，故此不再赘述。个人比较喜欢Debian，在安装进行到分区这一步时，可以根据个人喜好调整swap大小，后续可以在系统内开启zswap。创建Btrfs主分区，将它的挂载点设置在“/”下。完成安装。(建议使用DVD镜像，选择站点设置时不使用镜像站点则可以跳过软件包更新直接使用DVD镜像自带的包进行安装，避免安装时过多的等待)

接着进入系统，切换为root用户，为自己创建的用户添加sudo权限，并进行换源。此处不再赘述。

至此，我们的系统就初步配置好了。

Step 1. 终端美化与代理设置#

终端个人推荐使用zsh，主题使用powerlevel10k，插件仅需zsh-autosuggestions与zsh-syntax-highlighting即可。配置可参照:https://www.haoyep.com/posts/zsh-config-oh-my-zsh/ ↗

代理的设置建议如下新建 ~/scripts/proxy.sh,并在该脚本文件中复制以下代码,其中hostip和port按需更改:

#!/bin/sh
hostip=127.0.0.1
port=7890

PROXY_HTTP="http://${hostip}:${port}"

set_proxy(){
  export http_proxy="${PROXY_HTTP}"
  export HTTP_PROXY="${PROXY_HTTP}"

  export https_proxy="${PROXY_HTTP}"
  export HTTPS_proxy="${PROXY_HTTP}"

  export ALL_PROXY="${PROXY_SOCKS5}"
  export all_proxy=${PROXY_SOCKS5}

  git config --global http.https://github.com.proxy ${PROXY_HTTP}
  git config --global https.https://github.com.proxy ${PROXY_HTTP}

  echo "Proxy has been opened."
}

unset_proxy(){
  unset http_proxy
  unset HTTP_PROXY
  unset https_proxy
  unset HTTPS_PROXY
  unset ALL_PROXY
  unset all_proxy
  git config --global --unset http.https://github.com.proxy
  git config --global --unset https.https://github.com.proxy

  echo "Proxy has been closed."
}

test_setting(){
  echo "Host IP:" ${hostip}
  echo "Try to connect to Google..."
  resp=$(curl -I -s --connect-timeout 5 -m 5 -w "%{http_code}" -o /dev/null www.google.com)
  if [ ${resp} = 200 ]; then
    echo "Proxy setup succeeded!"
  else
    echo "Proxy setup failed!"
  fi
}

if [ "$1" = "set" ]
then
  set_proxy

elif [ "$1" = "unset" ]
then
  unset_proxy

elif [ "$1" = "test" ]
then
  test_setting
else
  echo "Unsupported arguments."
fi

bash

在你的.zshrc或者.bashrc中添加alias proxy=“source ~/scripts/proxy.sh”

使用时只需要在终端输入proxy set;proxy unset;proxy test.

至此,终端美化与代理设置就初步完成了

Step 2. 安装NVIDIA驱动#

你完全可以sudo apt install nvidia-driver来安装开源驱动但是我更推荐安装闭源驱动，如果有内核更新，记得要sudo apt install linux-headers-$(uname -r).

安装 apt GPG keyring 包，目的是获取 GPG 密钥

wget https://developer.download.nvidia.com/compute/cuda/repos/debian12/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb

bash

可以到此位置https://developer.download.nvidia.com/compute/cuda/repos/ ↗浏览具体发行版

apt 安装驱动

sudo apt update
sudo apt -y install nvidia-driver cuda-drivers

bash

或者

sudo apt install nvidia-driver-assistant
nvidia-driver-assistant

bash

此时会给出像下面的指引

Detected GPUs:
  NVIDIA GeForce RTX 4060 Laptop GPU - (pci_id 0x28E0)

Detected system:
  Debian GNU/Linux 13

Please copy and paste the following command to install the open kernel module flavour:
  sudo apt-get install -Vy nvidia-open

bash

按照指引

sudo apt-get install -Vy nvidia-open

bash

即可

至此,你已经几乎完成了环境的搭建。

Step 3. 安装python环境管理工具#

这个看个人品味,我推荐使用miniconda,venv或者mamba.镜像站使用tuna或者bfsu.

在此笔者假设读者熟悉上面三种的任意一种。

创建完虚拟的环境后,进入虚拟环境对pip进行换源。直接运行。

pip install torch

shell

如果在安装完torch后安装了大量nvidia-*的包,那么就可以放心了,你安装的是pytorch with GPU

你已经完成了深度学习环境搭建,立刻开始愉快的学习吧！