3.要在容器中使用gpu,需要使用nvidia-container-toolkits(最早使用nvidia-docker被弃用了,nvdia-docker2也不能用了好像,没有实际验证)
nvidia docker, nvidia docker2, nvidia container toolkits三者的区别
官方下载不能下
官方下载方式
使用中科大镜像下载
中科大源
然后重启docker服务
sudo systemctl restart docker
运行一个镜像
sudo docker run –gpus all --rm nvidia/cuda:10.0-base nvidia-smi
nvidia-container-toolkits国内源下载
最新推荐文章于 2025-03-10 17:56:06 发布