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等软件
- 终端美化
一行命令,开箱即用
可根据本地驱动选择对应的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 文件
📎 参考文章
- 作者:ziuch
- 链接:https://ziuch.com/article/Docker-with-GPU
- 声明:本文采用 CC BY-NC-SA 4.0 许可协议,转载请注明出处。