0. 前言
ubuntu下安装cuda、cuddn等NVIDIA机器学习、深度学习环境往往是在使用Linux中最头疼的配置步骤,同时,由于nvidia的cuda、cuddn版本众多,这让统一环境开发成为了难点。而Nvidia官方也认清楚了这一点,并基于docker开发了nvidia docker并且提供cuda镜像,这让我们完全不用考虑环境问题了
环境:
ubuntu20.04
docker 19.03
1. 安装docker
1.查看ubuntu内核
uname -r
2.安装docker
sudo apt-get install docker.io
3.查看版本
docker version 或 docker -v
4.启动
sudo systemctl unmask docker.service
sudo systemctl unmask docker.socket
sudo systemctl start docker.service
5.查看状态
sudo systemctl status docker
2. 修改docker源
sudo gedit /etc/docker/daemon.json
添加以下内容:(使用阿里源)
{
"registry-mirrors":[
"https://emasu4pd.mirror.aliyuncs.com",
"http://hub-mirror.c.163.com"
],
"runtimes": {
"nvidia": {
"path": "/usr/bin/nvidia-container-runtime",
"runtimeArgs": []
}
}
}