lxd 容器内的深度学习服务器环境配置

发布于:2025-06-30 ⋅ 阅读:(23) ⋅ 点赞:(0)

lxd 容器内的深度学习服务器环境配置

前言

lxd 是一个多用户容器,各个容器之间互不干扰,但宿主机的环境配置和容器内的环境配置过程略有不同,此处记录经过验证的 lxd 容器内深度学习服务器环境配置过程。

配置容器软件源

备份原来的源:

sudo mv /etc/apt/sources.list /etc/apt/sources.list.bak

将源的内容设置为清华镜像:

sudo apt install vim
sudo rm /etc/apt/sources.list
sudo vim /etc/apt/sources.list

将其中内容改为:

# 默认注释了源码镜像以提高 apt update 速度,如有需要可自行取消注释
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-updates main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-updates main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-backports main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-backports main restricted universe multiverse
deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-security main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-security main restricted universe multiverse

# 预发布软件源,不建议启用
# deb https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-proposed main restricted universe multiverse
# deb-src https://mirrors.tuna.tsinghua.edu.cn/ubuntu/ jammy-proposed main restricted universe multiverse 作者:贝拉的海 https://www.bilibili.com/read/cv17891624?spm_id_from=333.999.0.0 出处:bilibili

更新软件列表:

sudo apt update
sudo apt upgrade

设置 pip 清华源

sudo apt install python3-pip
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

安装容器显卡驱动

容器显卡驱动需与宿主机相同,此处宿主机的显卡驱动为 NVIDIA-Linux-x86_64-550.90.07.run ,输入以下命令安装显卡驱动:

sudo bash ./NVIDIA-Linux-x86_64-550.90.07.run --no-kernel-module

安装 CUDA

本次安装使用的 CUDA 版本是 sudo sh cuda_12.0.1_525.85.12_linux.run

sudo sh cuda_12.0.1_525.85.12_linux.run

不要勾选第一个安装显卡驱动( driver )的,因为之前已经安装过了:
在这里插入图片描述
安装完成后提示:
在这里插入图片描述

配置环境变量:

vim ~/.bashrc

再文件最后加入以下语句(如果安装的事别的版本的 CUDA,此处的 CUDA_HOME 要进行对于修改):

export CUDA_HOME=/usr/local/cuda-11.8
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}

然后使其生效:

source ~/.bashrc

使用命令 nvcc -V 查看安装的版本信息:
在这里插入图片描述

安装 miniconda

miniconda 下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/,本次使用的 miniconda 版本是 Miniconda3-py312_25.5.1-0-Linux-x86_64.sh
安装 miniconda :

sudo bash ./Miniconda3-py312_25.5.1-0-Linux-x86_64.sh

安装 pytorch

创建虚拟环境并进入虚拟环境:

conda create --name pytorch python=3.12  # 创建名为 pytorch 的虚拟环境,指定 Python 版本为 3.12

安装命令详见官网:https://pytorch.org/,并根据 CUDA 版本安装 pytorch ,此处是基于 CUDA 11.8 安装的 pytorch:

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

在命令号中输入 Python ,验证 pytorch 调用 GPU:

import torch
torch.cuda.is_available()

在这里插入图片描述


网站公告

今日签到

点亮在社区的每一天
去签到