Improve wrapdocker call method, make OSError not stop worker

Dockerfile-cuda

@@ -15,9 +15,6 @@ run apt-get update -qq && apt-get install -qqy \
     
 run curl -sSL https://get.docker.com/ | sh
 
-add ./wrapdocker /usr/local/bin/wrapdocker
-run chmod +x /usr/local/bin/wrapdocker
-
 volume /var/lib/docker
 
 env HF_HOME hf_home
@@ -33,4 +30,7 @@ run pip install -r requirements.cuda.txt
 run pip install -r requirements.txt
 run pip install -e .
 
-cmd ["wrapdocker"]
+add ./wrapdocker /usr/local/bin/wrapdocker
+run chmod +x /usr/local/bin/wrapdocker
+
+cmd ["wrapdocker", "bash"]

skynet/dgpu/network.py

@@ -27,7 +27,8 @@ async def failable(fn: partial, ret_fail=None):
 
     except (
         asks.errors.RequestTimeout,
-        json.JSONDecodeError
+        json.JSONDecodeError,
+        OSError
     ):
         return ret_fail
 

wrapdocker

@@ -108,6 +108,5 @@ else
 		fi
 		sleep 1
 	done
-	[[ $1 ]] && exec "$@"
+	[[ $1 ]] && exec bash -c "$@"
-	exec bash --login
 fi