Question

我只是使用TensorFlow来实现CNN模型。在训练过程中，有一个中间变量占用了很大的GPU内存，我想清除该变量的内存。

此变量称为'rgb_concat'，我只是尝试使用'rgb_concat = []'清除其内存，不确定是否在TensorFlow中有用吗？

如何在TensorFlow中实现这一目标？预先感谢！

一个名为“ rgb_concat”的中间变量，它占用较大的GPU内存，我想清除它，并将GPU内存保存给CNN模型中的其他层。如何在TensorFlow中实现它？

private void button3_Click(object sender, EventArgs e)
{ 
    using (SqlConnection conn = new SqlConnection(";Trusted_Connection=False"))
    using (SqlCommand comm = new SqlCommand())
    { 
            comm.Connection = conn;
            conn.Open(); 

            for (int i = 1; i < dataGridView2.Rows.Count; i++)
            {
                comm.CommandText = "INSERT INTO SOLine (CustID,Bookcode,Barcode,Description,Price,Disc,Qty) VALUES ('2058 KBC',@bookcode,@barcode,@desc,@price,@disc,@qty)";

                SqlParameter bookcode = comm.Parameters.AddWithValue("@bookcode", dataGridView2.Rows[i].Cells["BOOKCODE"].Value);
                SqlParameter barcode = comm.Parameters.AddWithValue("@barcode", dataGridView2.Rows[i].Cells["3"].Value);
                SqlParameter desc = comm.Parameters.AddWithValue("@desc", dataGridView2.Rows[i].Cells["4"].Value);
                SqlParameter price = comm.Parameters.AddWithValue("@price", dataGridView2.Rows[i].Cells["5"].Value);
                SqlParameter disc = comm.Parameters.AddWithValue("@disc", dataGridView2.Rows[i].Cells["6"].Value);
                //SqlParameter qty = comm.Parameters.AddWithValue("@qty", dataGridView2.Rows[i].Cells["7"].Value);

                if (dataGridView2.Rows[i].Cells["BOOKCODE"].Value == null) 
                {
                    bookcode.Value = DBNull.Value; 
                }

                if (dataGridView2.Rows[i].Cells["3"].Value == null) 
                {
                    barcode.Value = DBNull.Value; 
                }

                if (dataGridView2.Rows[i].Cells["4"].Value == null)
                { 
                    desc.Value = DBNull.Value; 
                }

                if (dataGridView2.Rows[i].Cells["5"].Value == null) 
                {
                    price.Value = DBNull.Value; 
                }

                if (dataGridView2.Rows[i].Cells["6"].Value == null)
                { 
                    disc.Value = DBNull.Value; 
                }

                // if (dataGridView2.Rows[i].Cells["7"].Value == null) 
                // { 
                //       qty.Value = DBNull.Value; 
                // }

                for (int q = 7; q <= dataGridView2.Columns.Count; q++) //dataGridView2.Columns.Count
                {
                    int w = 1;
                    w++;
                    comm.Parameters.Add("@qty", SqlDbType.Int).Value = dataGridView2.Rows[w].Cells[q].Value;
                    comm.Parameters.Clear();
                }

                comm.ExecuteNonQuery();
                comm.Parameters.Clear();
            } 

            MessageBox.Show("Save SQL"); 

            //try
            //{ 
            //    comm.ExecuteNonQuery();
            //}
            //catch (Exception ex)
            //{
            //    MessageBox.Show(ex.ToString());
            //}
        } 
    }

由于在第二个“ for”循环之后不再需要“ rgb_concat”，因此应在“ for”循环后将其清除。

Answer 1

您尝试过del关键字吗？

del rgb_concat

您也可以将变量设置为无。

rgb_concat = None

在TensorFlow中，如何清除CNN模型中中间变量的GPU内存？

1 个答案: