From dc296f64c65dfbb314bf3ae456e747e3b171cdb4 Mon Sep 17 00:00:00 2001
From: LXXXXR <73265258+LXXXXR@users.noreply.github.com>
Date: Sun, 16 May 2021 21:53:13 +0800
Subject: [PATCH] [Fix] Fix multi-node test tmp dir (#251)

* fix multi-node test tmp dir

* fix mmcv version
---
 mmcls/apis/test.py         | 3 ++-
 requirements/mminstall.txt | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/mmcls/apis/test.py b/mmcls/apis/test.py
index ec789b9ef..a7df59729 100644
--- a/mmcls/apis/test.py
+++ b/mmcls/apis/test.py
@@ -126,7 +126,8 @@ def collect_results_cpu(result_part, size, tmpdir=None):
                                 dtype=torch.uint8,
                                 device='cuda')
         if rank == 0:
-            tmpdir = tempfile.mkdtemp()
+            mmcv.mkdir_or_exist('.dist_test')
+            tmpdir = tempfile.mkdtemp(dir='.dist_test')
             tmpdir = torch.tensor(
                 bytearray(tmpdir.encode()), dtype=torch.uint8, device='cuda')
             dir_tensor[:len(tmpdir)] = tmpdir
diff --git a/requirements/mminstall.txt b/requirements/mminstall.txt
index d371e1cc8..521afb155 100644
--- a/requirements/mminstall.txt
+++ b/requirements/mminstall.txt
@@ -1 +1 @@
-mmcv-full>=1.3.0
+mmcv-full>=1.3.0,<1.5.0