ubuntu18.04从零开始离线搭建基于docker的pytorch环境

ubuntu18.04从零开始离线搭建基于docker的pytorch环境

  • 整体流程
    • 配置apt国内镜像源
    • 一、安装nvidia驱动
      • (一)禁用nouveau*
      • (二)离线安装编译环境gcc、make、build-essential*
      • (三)离线安装nvidia驱动*
    • 二、安装docker
      • (一)下载安装包
      • (二)安装*
      • (三)配置*
    • 三、安装nvidia-docker
      • (一)下载安装包
      • (二)安装*
      • (三)配置runtime*
    • 四、运行pytorch
      • (一)修改docker国内源
      • (二)拉取pytorch镜像并导出
      • (三) 导入镜像、测试*

任务需要搭建一个pytorch环境调用gpu资源,离线环境,踩坑无数,特地写下此文以备后用。

整体流程

一、安装nvidia驱动
二、安装docker
三、安装nvidia-docker
四、运行pytorch

除离线环境外,需一台联网ubuntu主机进行部分配置及安装包下载,以下步骤可先在联网主机上进行一次,再将安装包拷到离线主机上再进行一次,其中带*符号的步骤在离线主机上必须进行
文中涉及的所有deb安装包:docker、nvidia-docker、gcc、make、build-essential离线包.zip - CSDN下载

配置apt国内镜像源

若联网主机未修改apt镜像源,又不具备科学上网的条件,下载时速度将会非常令人感动:

  1. 备份sources.list(Ubuntu恢复文件很复杂,尽量避免直接删除系统文件):

     sudo cp /etc/apt/sources.list /etc/apt/sources.list.bak
    
  2. 修改sources.list:

     sudo gedit /etc/apt/sources.list
    

    替换阿里云镜像源:

     deb http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse
     deb http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse
     deb http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse
     deb http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse
     deb http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse
    
     deb-src http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse
     deb-src http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse
     deb-src http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse
     deb-src http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse
     deb-src http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse
    
  3. 更新

     sudo apt update
    

一、安装nvidia驱动

你可能感兴趣的:(教程,离线环境,docker,ubuntu,容器)