2.2.6. GPU专用¶
和普通不同的点是增加了 --gpus=all
:
docker run -it -gpus all <image> nvidia-smi
常见问题¶
Error response from daemon: could not select device driver “” with capabilities: [[gpu]]:
没有配置好 docker 运行环境导致的,执行安装容器运行环境
$ apt-get install nvidia-container-runtime
安装后重新启动 docker