亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

即使我在每次迭代結束時刪除所有變量,python for 循環中的內存泄漏

即使我在每次迭代結束時刪除所有變量,python for 循環中的內存泄漏

慕的地6264312 2023-03-30 10:10:20
以下兩個循環將使用內存,直到我用完,但我不明白為什么。我在每次迭代結束時刪除所有創建的變量,但它仍然泄漏。!pip3 install cupy-cuda101import cupy as cpimport numpy as npfrom sklearn.preprocessing import PolynomialFeaturesxtrain = cp.asnumpy(cp.random.uniform(-1,1,size = (150000,50)))for i in range(0,1000):   weights = cp.random.uniform(-1,1,size = (1275,1000))      for chunk in range(0,xtrain.shape[0],5000):      xchunk = xtrain[chunk:chunk+5000,:]      poly=PolynomialFeatures(interaction_only = True, include_bias = False)      xchunk = cp.array(poly.fit_transform(xchunk))      ranks = cp.matmul(xchunk,weights)      del ranks, xchunk, poly   del weightsxtrain 也只是浮點數據,介于 -1 和 1 之間。
查看完整描述

1 回答

?
蕭十郎

TA貢獻1815條經驗 獲得超13個贊

這些行在每次迭代結束時插入修復它:

    cp.get_default_memory_pool().free_all_blocks()
    cp.get_default_pinned_memory_pool().free_all_blocks()


查看完整回答
反對 回復 2023-03-30
  • 1 回答
  • 0 關注
  • 205 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號