nvidia-docker运行容器后找不到GPU

执行以下步骤:

1、保证主机已成功安装驱动,执行nvidia-smi查看。

2、创建容器时使用-e参数添加下面两个环境变量:

  NVIDIA_DRIVER_CAPABILITIES=compute,utility

  NVIDIA_VISIBLE_DEVICES=all

  完整命令为:

docker run -it -e NVIDIA_DRIVER_CAPABILITIES=compute,utility -e NVIDIA_VISIBLE_DEVICES=all $image /bin/bash

 

posted @ 2021-12-02 14:08  浅尝辄止~  阅读(2124)  评论(0编辑  收藏  举报