[Bugfix] Check file exists before removing it (#3178)

diff --git a/python/tvm/contrib/download.py b/python/tvm/contrib/download.py
index 46a9a6d988791b4a2444671610ace606fe08be32..bc6d32476eb72dd851d2cf5535ca3762257f9a7b 100644 (file)
--- a/python/tvm/contrib/download.py
+++ b/python/tvm/contrib/download.py
@@ -111,7 +111,8 @@ def download(url, path, overwrite=False, size_compare=False, verbose=1, retries=
         except Exception as err:
             retries -= 1
             if retries == 0:
-                os.remove(tempfile)
+                if os.path.exists(tempfile):
+                    os.remove(tempfile)
                 raise err
             else:
                 print("download failed due to {}, retrying, {} attempt{} left"