torch7 刪除cuda變數

2021-08-17 01:28:19 字數 586 閱讀 3838

經常會在torch上會有用到刪除變數的操作

a =nil
但是放到cuda上面卻不是很奏效

cutorch.setdevice(1)

src1 = torch.cudatensor(100000,10000,2)

這個時候會發現視訊記憶體占用了很大,但是怎麼來刪掉src1占用的視訊記憶體呢?

src1 = nil

collectgarbage();

發現這樣做不是很奏效,變數雖然變成了nil,可是占用的視訊記憶體還是沒有釋放,又不想kill掉執行緒重新來過,怎麼辦呢

!!!!!!!!!!!!!!!!

加上 thc_caching_allocator=0。

github上關於該問題的鏈結

再啟動torch的時候記得加這句話,例如

這個時候再講變數置為nil,collectgarbage就可以將占用的視訊記憶體清楚了

Torch7模型訓練

torch7搭建卷積神經網路詳細教程 已經詳細的介紹啦module模組,這裡再次基礎上再給出一些上container transfer functions layers和 layers模組的理解。並在後面給出一些簡單的模型訓練方法。下述程式在itorch qtconsole下執行。主要有四個函式 詳...

ubuntu安裝Torch7教程

安裝torch7很簡單,只需要幾條命令即可,本教程在安裝完torch7之後還會download下demo,並執行其中的train a digit classifier。具體安裝步驟如下 sudo apt get install git git clone torch recursive cd tor...

Torch7深度學習教程(二)

這是自己定義tensor的值輸入10個整型值,列印price,注意這時它只是乙個有10資料的tensor。對比這兩張,第一張是有10個資料的tensor,第二張是10x1的矩陣,也可以成為列向量 神經網路經常用到 是呼叫了reshape 方法轉換的,它們看著一樣本質上是不一樣的!這時再次呼叫resh...