nvidia-docker部署pytorch服务【GPU工作站】

文章目录

  • 一、安装 Docker
  • 二、安装 NVIDIA Container Toolkit
  • 三、宿主机安装 cuda 和 nvidia-driver
  • 四、测试


一、安装 Docker

可以参考这篇文章
https://blog.csdn.net/weixin_43721000/article/details/124237932


二、安装 NVIDIA Container Toolkit

参考nvidia官方
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html

  1. 安装 nvidia-docker

    • centos安装方法

      1. Configure the repository

你可能感兴趣的:(python高级,服务部署,Docker,docker,pytorch,容器)