容器内输入nvidia-smi出现CUDA Version:N/A问题

首先定位问题，宿主机cuda显示正常
容器内驱动显示正常，cuda不显示。
解决方案：
需要在启动容器时候加上
-e NVIDIA_DRIVER_CAPABILITIES=compute,utility
再进入容器，可以使用nvidia-smi看到cuda版本

继续验证，安装pytorch

import torch
print(torch.version.cuda)
print(torch.backends.cudnn.version())
print(torch.cuda.is_available())

可以看到已经成功加载cuda

解决！！！