docker gpu容器镜像环境构建

1、下载nvidia官方的环境镜像

https://hub.docker.com/r/nvidia/cuda

docker pull nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04

2、安装python环境

进入容器

docker run -it   --gpus all   -v /mnt/sdc1/loong/gm_tts:/data  ***bash

安装python

##启用 universe 仓库
apt update
apt install -y software-properties-common
add-apt-repository universe
apt update
##装 Python 3.10
apt install -y \
  python3.10 \
  python3.10-venv \
  python3.10-dev

apt update
apt install -y python3-pip