背景: 安装好docker和nvidia-docker以后,需要测试容器里面GPU是否可用:

命令1:

sudo docker run --runtime=nvidia --rm nvidia/cuda nvidia-smi

命令2:

sudo docker run --gpus all --rm nvidia/cuda nvidia-smi

有时候第一个命令不好使,可以测试一下第二个命令

reference :
[1] https://github.com/NVIDIA/nvidia-docker/issues/838

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐