Cuda 11/10的Dockerfile
10 11 CUDA Dockerfile
2023-09-14 09:15:32 时间
说明:如果NVIDIA显卡不满足要求,请修改dockerfile文件配置要求或慎重使用,关于dockerfile使用请参考我的博客文章:
https://blog.csdn.net/weixin_41194129/category_10210023.html
cuda11相关链接如下:
https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/
https://github.com/triton-inference-server/server/releases/tag/v2.7.0
https://github.com/Algorithm-learning-community-for-python/project_dockerfile/tree/master/cuda11
https://github.com/EnzoDAndrea/Container-TF2.4-CUDA-11
CUDA10.1-ubuntu16.04:
安装Docker 参考ubuntu安装docker
安装NVIDIA Container Toolkit 参考NVIDIA/nvidia-docker
准备好Python-3.6.9.tar.xz
从nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04 镜像基础上搭建
安装openssh-server、python、pytorch
run镜像时加上参数--gpus all --ipc=host
# BASE IMAGE
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
# LABEL MAINTAINER
LABEL xxx="xxx@xxx"
SHELL ["/bin/bash","-c"]
WORKDIR /tmp
# copy安装文件
COPY Python-3.6.9.tar.xz /tmp
# 设置 root 密码
RUN echo 'root:password' | chpasswd \
# 安装openssh-server 并配置
&& apt-get update && apt-get -y install openssh-server \
&& sed -i 's/UsePAM yes/UsePAM no/g' /etc/ssh/sshd_config \
&& sed -i 's/PermitRootLogin prohibit-password/PermitRootLogin yes/g' /etc/ssh/sshd_config \
&& mkdir /var/run/sshd \
# 安装python依赖包
&& apt-get -y install build-essential python-dev python-setuptools python-pip python-smbus \
&& apt-get -y install build-essential libncursesw5-dev libgdbm-dev libc6-dev \
&& apt-get -y install zlib1g-dev libsqlite3-dev tk-dev \
&& apt-get -y install libssl-dev openssl \
&& apt-get -y install libffi-dev \
# 安装python 3.6.9
&& mkdir -p /usr/local/python3.6 \
&& tar xvf Python-3.6.9.tar.xz \
&& cd Python-3.6.9 \
&& ./configure --prefix=/usr/local/python3.6 \
&& make altinstall \
# 建立软链接
&& ln -snf /usr/local/python3.6/bin/python3.6 /usr/bin/python3 \
&& ln -snf /usr/local/python3.6/bin/pip3.6 /usr/bin/pip3\
# 安装pytorch
&& mkdir ~/.pip && echo -e '[global] \nindex-url = https://mirrors.aliyun.com/pypi/simple/' >> ~/.pip/pip.conf \
&& pip3 install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html \
# 清理copy的安装文件
&& apt-get clean \
&& rm -rf /tmp/* /var/tmp/*
EXPOSE 22
CMD ["/usr/sbin/sshd", "-D"]
构建镜像命令:
docker build -t dockerfile .
相关文章
- 10个.htaccess在wp的技巧
- 【说站】TIOBE 10月编程语言排行榜,Python位临榜首
- 2022-10-11:一个整数区间 [a, b] ( a < b ) 代表着从 a 到 b 的所有连续整数,包括 a 和 b。 给你一组整数区间interval
- 已知如下数组: var arr = [ [1, 2, 2], [3, 4, 5, 5], [6, 7, 8, 9, [11, 12, [12, 13, [14] ] ] ], 10];
- 2022-10-15-整洁代码的注释与格式
- 测试人必备的10款实用谷歌插件,压箱分享!
- 如何在Windows 10的VirtualBox中安装macOS High Sierra
- Ableton Live 11 Suite for Mac(音乐制作软件) v11.2.10/v10.1.43中文激活版
- MySQL 8.0.31于10月11日正式发布,更新内容速览
- C++面向对象程序设计(谭浩强)第三章第10~11节学习笔记
- SQLSever查询第11行到第25行的数据,sql分页查询 ( offset 10 rows fetch next 15 rows only)详解数据库
- 深入了解Oracle 10数据库(oracle10数据库)
- 微软再次确认10月发布Windows 11 RTM版本
- Windows 10 VS. Linux
- 写给 Python 开发者的 10 条机器学习建议
- 大优点10大关于Linux的优点:让你了解其以及它的能力(linux的10)
- 如何实施Oracle 10的升级方案(oracle10升级方案)
- 比5G还要快10倍,这项技术可能在2020年问世
- Daydream View VR头盔将于11月10日发售,都有这些可玩的