type
status
date
slug
summary
tags
category
icon
password
Last edited time
Mar 16, 2024 07:01 AM
😀
基于Nvidia官方镜像构建包含必要软件的Docker镜像,并创建相应容器进行深度网络的训练

📝 主旨内容

镜像内容

  • 安装当前CUDA版本下最新版本的Pytorch
  • 安装miniconda并更换为清华源
  • ubuntn软件源更换为清华源
  • pip更换成中科大源
  • 安装vim,tmux,wget,git,unzip,sudo等软件
  • 终端美化
    • notion image

一行命令,开箱即用

💡
可根据本地驱动选择对应的cuda版本进行安装,本地运行nvidia-smi 查看最高支持的cuda版本(可选有cuda-12.1, cuda-11.8, cuda-11.6)

容器的操作

1. 进入容器

💡
也可以通过SSH工具连接进入,注意端口是8022,用户是ubuntu,密码是123(需要先进入容器启动ssh服务 sudo service ssh start)

2. 本地-容器文件拷贝

3. 退出容器

4. 打包容器为镜像文件

4.1 基于容器创建镜像
4.2 将镜像导出成文件

🤗 总结归纳

以下是通过Dockerfile构建镜像
以下压缩包包含全部所需文件,点击即可下载

运行脚本生成镜像(cuda:11.6.1-cudnn8-devel-ubuntu20.04)

💡
当前目录下需存在Dockerfile文件.condarc文件.tmux.conf文件.vimrc文件

镜像的创建(上面四个文件的内容 可选修改)

Dockerfile文件

.condarc文件

.tmux.conf 文件

.vimrc 文件

📎 参考文章

 
曹云康——华中科技大学2020级博士Docker常用命令
Loading...