- create a new notebook
- choose python3
- choose [runtime] --> [change runtime type]
- choose GPU
Tensorflow OK (installed originally)
Keras OK
PyTorch OK
Chainer OK after 4.0.0b4
MXNet OK
Caffe NA
CNTK NA
Sonnet OK
install keras
!pip install keras
pytorch
!pip install http://download.pytorch.org/whl/cu80/torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl
!pip install torchvision
chainer
!apt-get install -y -qq libcusparse8.0 libnvrtc8.0 libnvtoolsext1
!ln -snf /usr/lib/x86_64-linux-gnu/libnvrtc-builtins.so.8.0 /usr/lib/x86_64-linux-gnu/libnvrtc-builtins.so
!pip install cupy-cuda80==4.0.0b4
!pip install chainer==4.0.0b4
mxnet
!pip install mxnet-cu80
sonnet
!pip install dm-sonnet
cupy
!apt -y install libcusparse8.0 libnvrtc8.0 libnvtoolsext1
!ln -snf /usr/lib/x86_64-linux-gnu/libnvrtc-builtins.so.8.0 /usr/lib/x86_64-linux-gnu/libnvrtc-builtins.so
!pip install 'cupy-cuda80==4.0.0b4' 'chainer==4.0.0b4'