1.1 本地安装 docker 工具
这里以ubutun下安装docker为例,其他操作系统安装命令略有不同,可自行百度。(建议使用阿里源安装速度快)
sudo apt install docker.io
如果你本地有gpu,请继续执行如下命令以支持gpu调用:
注意:
英伟达对 docker 支持的 linux 发行版:https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/supported-platforms.html
**英伟达目前支持 Ubuntu 最高的版本是 22.04,最新的 Ubuntu-24.04 是不支持的,所以暂时不要安装 Ubuntu-24.04 的版本 **
# Add the package repositories
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker
docker在新的版本中均已支持直接调用gpu,通过–gpu 指定使用哪个gpu, --gpu all 则是使用所有gpu 前提:请确保已按照前文环境篇linux末尾安装了Nvidia对docker的软件支持。
深度学习常用镜像集合(包含国内源和海外源)
https://tianchi.aliyun.com/forum/post/67720