This commit is contained in:
2024-02-20 09:54:12 +08:00
parent 2139b10fc6
commit 0bfa4f7b04
2 changed files with 21 additions and 1 deletions

View File

@@ -77,6 +77,7 @@ curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-contai
| sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
sudo apt-get update
sudo apt-get install -y nvidia-container-toolkit
sudo apt-get install nvidia-container-runtime -y
```
Configure Docker to use Nvidia driver
@@ -86,6 +87,26 @@ sudo nvidia-ctk runtime configure --runtime=docker
sudo systemctl restart docker
```
实际上这个操作修改了/etc/docker/daemon.json文件
```shell
{
"registry-mirrors": [
"https://0b27f0a81a00f3560fbdc00ddd2f99e0.mirror.swr.myhuaweicloud.com",
"https://ypzju6vq.mirror.aliyuncs.com",
"https://registry.docker-cn.com",
"http://hub-mirror.c.163.com",
"https://docker.mirrors.ustc.edu.cn"
],
"runtimes": {
"nvidia": {
"args": [],
"path": "nvidia-container-runtime"
}
}
}
```
配置完成之后:
```shell

View File

@@ -74,7 +74,6 @@ c.DockerSpawner.extra_create_kwargs.update({
# 'JUPYTERHUB_CORS': '{"allow_origin_pat": "https?://.*\\.example\\.com"}'
# }
# GPU 和网络配置
c.DockerSpawner.extra_host_config = {
'runtime': 'nvidia'