diff --git a/Dockerfile b/Dockerfile index 7f475b31b64f0..a97e4849b3031 100644 --- 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