반응형
FROM ubuntu:latest
MAINTAINER Jongyeol Yang <yangjy0113@gmail.com>
# basic apt
RUN apt-get update
RUN apt-get install -y build-essential vim wget
# CUDA Repository
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_6.5-14_amd64.deb
RUN dpkg -i cuda-repo-ubuntu1404_6.5-14_amd64.deb
RUN apt-get update
# CUDA Toolkit
RUN apt-get install cuda
# Environment Variables
RUN export CUDA_HOME=/usr/local/cuda-6.5
RUN export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
RUN PATH=${CUDA_HOME}/bin:${PATH}
RUN export PATH
# CUDA SDK Samples
RUN cuda-install-samples-6.5.sh ~
RUN cd ~/NVIDIA_CUDA-6.5_Samples; make
반응형
'IT > DeepLearning' 카테고리의 다른 글
AlphaCode (0) | 2022.02.08 |
---|---|
LaMDA (구글에서 발표한 오픈 도메인 챗봇) (0) | 2022.02.04 |
Dockerfile(theano) (0) | 2014.11.11 |