用dockerfile 安装 CUDA 7.0, CuDNN 5.0 and Ubuntu 14.04
时间: 2024-09-08 09:02:58 浏览: 136
要使用Dockerfile安装CUDA 7.0、CuDNN 5.0以及基于Ubuntu 14.04的环境,你需要遵循以下步骤。首先,你需要从NVIDIA官方获取CUDA和CuDNN的安装包。请注意,NVIDIA官方并不直接提供Docker镜像,因此你需要从官方下载安装包并按照NVIDIA的指导手册进行安装。
以下是一个简单的Dockerfile示例,用于创建一个安装了CUDA 7.0和CuDNN 5.0的Ubuntu 14.04环境:
```Dockerfile
# 使用Ubuntu 14.04作为基础镜像
FROM ubuntu:14.04
# 更新系统并安装必要的依赖
RUN apt-get update && apt-get install -y \
build-essential \
libtool \
libjpeg8-dev \
libjasper-dev \
libpng12-dev
# 设置环境变量
ENV CUDA_VERSION 7.0
ENV CUDNN_VERSION 5.0
# 添加NVIDIA驱动安装脚本(假设脚本已经下载到本地)
ADD nvidia驱动安装脚本.sh /usr/local/bin/
# 安装NVIDIA驱动
RUN sh /usr/local/bin/nvidia驱动安装脚本.sh && \
rm /usr/local/bin/nvidia驱动安装脚本.sh
# 添加CUDA 7.0安装包(假设包已经下载到本地)
ADD cuda-repo-ubuntu1404_7.0-28_amd64.deb /tmp/
# 安装CUDA 7.0
RUN dpkg -i /tmp/cuda-repo-ubuntu1404_7.0-28_amd64.deb && \
apt-get update && \
apt-get install -y cuda=${CUDA_VERSION}-1 && \
rm /tmp/cuda-repo-ubuntu1404_7.0-28_amd64.deb && \
rm -rf /var/lib/apt/lists/*
# 添加CuDNN 5.0安装包(假设包已经下载到本地)
ADD libcudnn5_5.0.50-1+cuda7.0_amd64.deb /tmp/
ADD libcudnn5-dev_5.0.50-1+cuda7.0_amd64.deb /tmp/
# 安装CuDNN 5.0
RUN dpkg -i /tmp/libcudnn5_5.0.50-1+cuda7.0_amd64.deb /tmp/libcudnn5-dev_5.0.50-1+cuda7.0_amd64.deb && \
rm /tmp/libcudnn5_5.0.50-1+cuda7.0_amd64.deb /tmp/libcudnn5-dev_5.0.50-1+cuda7.0_amd64.deb
# 设置环境变量
ENV PATH /usr/local/cuda-${CUDA_VERSION}/bin:${PATH}
ENV LD_LIBRARY_PATH /usr/local/cuda-${CUDA_VERSION}/lib64:${LD_LIBRARY_PATH}
# 安装Python 2.7 (可选)
RUN apt-get install -y python2.7
# 安装Python开发包 (可选)
RUN apt-get install -y python2.7-dev
# 清理缓存和临时文件
RUN apt-get clean && \
rm -rf /var/lib/apt/lists/*
# 其他自定义命令和设置
```
在构建这个Docker镜像之前,你需要从NVIDIA官方网站下载CUDA和CuDNN的安装包,并且编写一个安装NVIDIA驱动的脚本。然后,将这些文件放置在Dockerfile可以访问的目录中。
请注意,由于NVIDIA的安装程序可能需要交互式安装,直接在Dockerfile中执行可能会遇到问题。因此,通常的做法是在构建镜像之前先手动安装驱动和CUDA,然后将其打包成一个新的基础镜像。
阅读全文