use cpu

5bd136ae · 赵威 · b0a71fc8 · 5bd136ae
Commit 5bd136ae authored Jul 23, 2020 by 赵威
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

main.py src/main.py +2 -0

No files found.
--- a/src/main.py
+++ b/src/main.py
@@ -23,6 +23,8 @@ def main():

    tf.logging.set_verbosity(tf.logging.INFO)

+    os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+
    device_df, diary_df, click_df, conversion_df = read_csv_data(Path("~/data/cvr_data/"))
    # print(diary_df.sample(1))
    device_df = device_feature_engineering(device_df)