Skip to content
Permalink
Browse files

Merge pull request #8 from chainer/mitmul-patch-1

Support CUDA 10.0 container
  • Loading branch information...
kmaehashi committed Feb 6, 2019
2 parents 1458c70 + bc5efb8 commit 62bfc1a40058595775b06d122166fbb9fa45e831
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 62bfc1a

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