reduce the size of Docker image (pytorch#1729)

CookieMonsterYan · Jun 5, 2017 · 3716286 · 3716286
1 parent c357ebd
commit 3716286
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -30,7 +30,7 @@ COPY . .
 
 RUN TORCH_CUDA_ARCH_LIST="3.5 5.2 6.0 6.1+PTX" TORCH_NVCC_FLAGS="-Xfatbin -compress-all" \
     CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" \
-    python setup.py install
+    pip install -v .
 
 WORKDIR /workspace
 RUN chmod -R a+w /workspace