No module named 'cupy' on Google Colab

Question

I'm working on the FastPhotoStyle project: https://github.com/NVIDIA/FastPhotoStyle

and I follow the steps of its tutorial: https://github.com/NVIDIA/FastPhotoStyle/blob/master/TUTORIAL.md

I'm running Example 1 on Google Colab where default environment is

• CUDA 10.0
• Python 3.6
• Chainer 5.4.0
• CuPy 5.4.0

This is how I tried on Colab Notebook:

https://colab.research.google.com/drive/1oBgdJgXCLlUQhpwPoG1Uom3OKTzHR4BF

After running

!python3 demo.py --output_image_path results/example1.png

Here's the error message I got:

Traceback (most recent call last): File "demo.py", line 9, in

import process_stylization

File "/content/drive/FastPhotoStyle/process_stylization.py", line 14, in

from smooth_filter import smooth_filter

File "/content/drive/FastPhotoStyle/smooth_filter.py", line 327, in

from cupy.cuda import function

ModuleNotFoundError: No module named 'cupy'

Could someone help me with it please?

Answer 1

import time 
import cupy as cp
import numpy as np
s = time.time()
A = np.random.randint(10, size=(1000,1000))
C = A.transpose()
e = time.time()
print("waktu CPU:",e - s, "detik")

s = time.time()
A = cp.random.radint (10,size=(1000,1000))
C = A. transpose()
cp.cuda.Stream.null.synchronize()`enter code here`
e = time.time()
print("waktu GPU:",e - s, "detik")

Answer 2

have you tried running the demo with python full path?

!/usr/local/lib/python3.6//bin/python3 demo.py --output_image_path 

or running without anaconda installation by simple install all necessary libraries

!pip install cupy

or see installation instructions here