Skip to content
Permalink
Browse files

Support CUDA 10.0 container

  • Loading branch information...
mitmul committed Feb 6, 2019
1 parent 1458c70 commit bc5efb89622d9db341ff1833f1d1fec37080e6b6
Showing with 7 additions and 1 deletion.
  1. +7 −1 docs/install
@@ -17,7 +17,13 @@ _EOF_
fi


if [ -d /usr/local/cuda-9.2 ]; then
if [ -d /usr/local/cuda-10.0 ]; then
# For CUDA 10.0 container
set -ex
apt -y -q install cuda-libraries-dev-10-0
pip install -q "cupy-cuda100 ${CUPY_VERSION}" "chainer ${CHAINER_VERSION}"

elif [ -d /usr/local/cuda-9.2 ]; then
# For CUDA 9.2 container
set -ex
apt -y -q install cuda-libraries-dev-9-2

0 comments on commit bc5efb8

Please sign in to comment.
You can’t perform that action at this time.