Skip to content
GitLab
Menu
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
Viacheslav Ostroukh
research-docker
Commits
3a126dde
Commit
3a126dde
authored
Jun 07, 2018
by
Viacheslav Ostroukh
🚲
Browse files
add Dockerfile for CUDA research container
parent
d02d37ea
Pipeline
#10010
passed with stage
in 17 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
cuda/Dockerfile
0 → 100644
View file @
3a126dde
FROM
quantumtinkerer/research:latest
USER
root
WORKDIR
/
# FIXME 9.2 is hardcoded along the Dockerfile
ENV
CUDA_VERSION 9.2.88
ENV
CUDA_PKG_VERSION 9-2=${CUDA_VERSION}-1
# For now CUDA is not supported on 18.04, experimental image of NVidia uses this
# one
RUN
curl
-fsSL
https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1710/x86_64/7fa2af80.pub | apt-key add -
\
&&
echo
"deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1710/x86_64 /"
>
/etc/apt/sources.list.d/cuda.list
\
&&
apt-get update
\
&&
apt-get
install
-y
--no-install-recommends
\
cuda-libraries-
$CUDA_PKG_VERSION
\
cuda-libraries-dev-
$CUDA_PKG_VERSION
\
cuda-nvtx-
$CUDA_PKG_VERSION
\
cuda-minimal-build-
$CUDA_PKG_VERSION
\
cuda-command-line-tools-
$CUDA_PKG_VERSION
\
cuda-cudart-
$CUDA_PKG_VERSION
\
&&
apt-get clean
\
&&
ln
-s
/usr/local/cuda-9.2 /usr/local/cuda
\
&&
rm
-rf
/var/lib/apt/lists/
*
RUN
echo
"/usr/local/nvidia/lib"
>>
/etc/ld.so.conf.d/nvidia.conf
&&
\
echo
"/usr/local/nvidia/lib64"
>>
/etc/ld.so.conf.d/nvidia.conf
ENV
PATH /usr/local/cuda/bin:${PATH}
ENV
NVIDIA_VISIBLE_DEVICES all
ENV
NVIDIA_DRIVER_CAPABILITIES compute,utility
ENV
NVIDIA_REQUIRE_CUDA "cuda>=9.2"
USER
$NB_UID
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment