google colaboratoryにてcupyを使用した計算を行っているのですが、forループを実行していると、ある回数を超えると数式の実行速度が0.1~0.5ms程度から10msほどに低下します。
メモリ不足かと思ったのですが、GPUメモリは宣言した行列に比べて余裕があります。
なぜ実行速度が低下してしまうのでしょうか。
実行環境とコードは以下となります。
他に必要な情報等ございましたらご返信ください。
どうぞよろしくお願いいたします。
kankyou
1OS : Ubuntu 18.04.5 LTS 2CPU: Intel(R) Xeon(R) CPU @ 2.20GHz 2コア 3Mem: 12GB 4GPU: Tesla P100-PCIE 16GB
python
1import time 2import cupy as cp 3cdot = cp.dot 4cdiff = cp.diff 5cmultiply = cp.multiply 6csubtract = cp.subtract 7cfloat32 = cp.float32 8cfloat64 = cp.float64 9ccopy = cp.copy 10cadd = cp.add 11 12a = 1500 13b = 1000 14 15c = a*b 16d = a/b 17e = 0.001 18 19x = 10000 20y = 10000 21print(x,y) 22avet = 0 23t1 = 0 24t2 = 0 25time1 = 0 26N = 1000 27 28DIFF = cp.ones((x+1,y+1),dtype=cfloat32) 29ANS = cp.zeros((x+1,y+1),dtype=cfloat32) 30coef1 = cp.zeros((x+1,y),dtype=cfloat32) 31coef2 = cp.zeros((x+1,y),dtype=cfloat32) 32 33coef1 = (a*c*e) + coef1 34coef2 = (b*d*e) + coef2 35 36for loop in range(N): 37 t1 = time.time() 38 ANS[:,0:y] = cadd(cmultiply(cdiff(DIFF,n=1,axis=1),coef1),\ 39 cmultiply(cdiff(DIFF,n=1,axis=1),coef2),out=ANS[:,0:y]) 40 ANS[:,0:1] = 0.0 41 ANS[:,y:y+1] = 0.0 42 t2 = time.time() 43 print(t2-t1) 44 time1 += t2-t1 45avet = time1 / N 46print(avet)```
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。