refactor(//docker): Make it easier to run applications in the repo with

the container

Signed-off-by: Naren Dasan <[email protected]
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

core/execution/register_trt_op.cpp

@@ -39,7 +39,6 @@ std::vector<at::Tensor> execute_engine(std::vector<at::Tensor> inputs, c10::intr
         LOG_DEBUG("Output shape: " << out_shape);
         auto dims = core::util::toVec(out_shape);
         auto type = util::toATenDType(compiled_engine->exec_ctx->getEngine().getBindingDataType(o));
-        std::cout << pyt_idx << std::endl;
         outputs[pyt_idx] = std::move(at::empty(dims, {at::kCUDA}).to(type).contiguous());
         gpu_handles.push_back(outputs[pyt_idx].data_ptr());
     }

docker/Dockerfile.20.07

@@ -33,4 +33,5 @@ RUN conda init bash
 
 ENV LD_LIBRARY_PATH /opt/conda/lib/python3.6/site-packages/torch/lib:$LD_LIBRARY_PATh
 
-WORKDIR /opt/trtorch/notebooks
+WORKDIR /opt/trtorch/
+CMD /bin/bash