k8s使用寒武纪显卡
时间: 2023-11-08 09:54:50 浏览: 162
寒武纪DianNao系列论文
5星 · 资源好评率100%
k8s可以使用寒武纪显卡进行加速计算。通过在容器中执行相应的命令,可以指明使用哪几张卡。例如,可以使用以下命令来运行容器并使用寒武纪显卡:
```
docker run --gpus '"device=1,2"' nvidia/cuda nvidia-smi
```
这个命令会在容器内输出宿主机上的Nvidia显卡数量,以确保模型能够使用到显卡加速。另外,如果你使用的是pytorch,可以在容器内执行以下代码来检查显卡是否可用:
```
import torch
torch.cuda.is_available()
```
阅读全文