IT/DeepLearning

Dockerfile(cuda-65)

알 수 없는 사용자 2014. 11. 11. 17:13
반응형

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