gpu

1、检测是否有gpu卡：lspci|grep -i nvidia

2、gpu驱动是否已经安装：which nvidia-smi

3、安装nvidia运行时：https://github.com/NVIDIA/nvidia-container-runtime

 

 

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.repo | \
sudo tee /etc/yum.repos.d/nvidia-container-runtime.repo

yum install nvidia-container-runtime

"default-runtime": "nvidia"


============ 安装nvidia-docker2
https://zhuanlan.zhihu.com/p/120047324

https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker