ubuntu docker炼丹环境配置

给大家推荐一个可以使用知识库和自己提示词的gpt网站超级好用：ITGPT

正题开始：

安装docker

sudo apt update
sudo apt install vim curl python3-pip git  -y
sudo curl -SsL get.docker.com | sudo bash

docker 换源

sudo mkdir -p /etc/docker
sudo tee /etc/docker/daemon.json <<-'EOF'
{
  "registry-mirrors": ["http://hub-mirror.c.163.com"]
}
EOF
sudo systemctl enable docker
sudo systemctl start docker
sudo groupadd docker
sudo gpasswd -a ${USER} docker
sudo chmod a+rw /var/run/docker.sock

docker run命令参考下面

docker run命令参考

NVIDIA runtime

Migration Notice | nvidia-container-runtime

curl -s -L https://nvidia.github.io/nvidia-container-runtime/gpgkey | \
  sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.list | \
  sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
sudo apt-get update

NVIDIA/nvidia-container-runtime: NVIDIA container runtime (github.com)

sudo apt-get install nvidia-container-runtime

docker NVIDIA显卡镜像

docker pull nvidia/cuda:11.8.0-devel-ubuntu22.04

cuda 11.7

docker pull nvidia/cuda:11.8.0-devel-ubuntu22.04


pip3 install torch torchvision torchaudio

cuda 11.6

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116
docker run -p 8000:8000 -d --gpu all --name jupyterhubgpu  --restart=always jockerdragon/jupyterhub:custoner-0.1