publish

.dockerignore

@@ -1,2 +1,28 @@
 Dockerfile
-**/publish.py
+**/publish.py
+my
+.git
+.refresh
+__pycache__
+.ipynb_checkpoints/
+.vscode/
+__res/
+perf.data
+perf.data.old
+*.swp
+*.ipynb
+*.pdf
+*.zip
+*.tgz
+test.py
+extern/mkl/mkldnn_lnx*/*
+data/
+build/
+venv/
+*.md
+!*.src.md
+!README.md
+!README.cn.md
+python/jittor.egg-info
+dist/
+!doc/source/*

Dockerfile

@@ -38,12 +38,14 @@ RUN pip3 install matplotlib
 
 RUN apt install openmpi-bin openmpi-common libopenmpi-dev -y
 
+RUN pip3 install jittor --timeout 100 && python3.7 -m jittor.test.test_example
+
+RUN pip3 uninstall jittor -y
+
 COPY . .
 
 RUN pip3 install . --timeout 100
 
 RUN python3.7 -m jittor.test.test_example
 
-RUN rm -rf ~/.cache/jittor/default
-
 CMD python3.7 -m jittor.notebook --allow-root --ip=0.0.0.0

python/jittor/utils/publish.py

@@ -25,8 +25,8 @@ def docker_task(name, build_cmd):
     run_cmd(build_cmd)
     run_cmd(f"sudo docker push {name}")
     bname = os.path.basename(name)
-    run_cmd(f"docker save {name}:latest -o /tmp/{bname}.tgz && chmod 666 /tmp/{bname}.tgz")
-    upload_file(f" /tmp/{bname}.tgz")
+    run_cmd(f"sudo docker save {name}:latest -o /tmp/{bname}.tgz && sudo chmod 666 /tmp/{bname}.tgz")
+    upload_file(f"/tmp/{bname}.tgz")
 
 docker_task(
     "jittor/jittor",