vast.ai compatability updates (#1657)

Dockerfile

@@ -1,7 +1,10 @@
 # Start FROM Nvidia PyTorch image https://ngc.nvidia.com/catalog/containers/nvidia:pytorch
 FROM nvcr.io/nvidia/pytorch:20.10-py3
 
-# Install dependencies
+# Install linux packages
+RUN apt install screen
+
+# Install python dependencies
 RUN pip install --upgrade pip
 # COPY requirements.txt .
 # RUN pip install -r requirements.txt

utils/google_utils.py

@@ -53,7 +53,7 @@ def attempt_download(weights):
             return
 
 
-def gdrive_download(id='1n_oKgR81BJtqk75b00eAjdv03qVCQn2f', name='coco128.zip'):
+def gdrive_download(id='1uH2BylpFxHKEGXKL6wJJlsgMU2YEjxuc', name='tmp.zip'):
     # Downloads a file from Google Drive. from utils.google_utils import *; gdrive_download()
     t = time.time()