测试docker里面的GPU是否可用
**背景:**安装好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有时候第一个命令不好使,可以测试一下第二个命...
·
背景: 安装好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
更多推荐
已为社区贡献3条内容
所有评论(0)