使用docker从零构建python、cuda、pytorch等环境

一、docker和nvidia-docker安装

本机先安装docker，然后安装nvidia-docker（在docker中使用gpu需要安装nvidia-docker，同时本机需要有nvidia显卡驱动，且该驱动版本要支持docker里面的cuda版本）。

本机安装nvidia-docker后，不需要再在docker容器中安装nvidia显卡驱动，便可使用gpu（另外有一种说法，可以直接在docker容器中安装nvidia显卡驱动，便可直接在docker容器使用gpu，不需要本机安装nvidia-docker，不过本人没有尝试过）。

安装过程官网教程参考：

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

二、拉取nvidia/cuda镜像，创建nvidia/cuda容器，在容器中配置python，pytorch等基本环境

1、拉取nvidia/cuda镜像后，手动配置相应环境，这种方法比较灵活。

另一种方法就是拉取带cuda环境pytorch的官方镜像，这里先讲第一种方法。

# 拉取nvidia-cuda官方镜像，查看相应版本镜像，官方地址：