最近、tx2 に CUDA 10.0 と cupy をインストールしました。cupy は numpy よりも 10 倍高速であるという記事を読みました。だから私はこのプログラムを実行しました:
### Numpy and CPU
s = time.time()
x_cpu = np.ones((100,100,100))
e = time.time()
print(e - s)
### CuPy and GPU
s = time.time()
x_gpu = cp.ones((100,100,100))
cp.cuda.Stream.null.synchronize()
e = time.time()
print(e - s)
私が得た結果は非常に異なっています。Numpy は 0.00273 しかかからなかったのに対し、Cupy は 0.6795 秒かかりました。キューピーがこんなに遅くなる原因は何ですか。次の方法でcupyをインストールしました:
sudo pip3 install cupy -vvv
どんな手がかりも役に立ちます。前もって感謝します!