fix(//py): Constrain the CUDA version in container builds

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

Author: narendasan

Diff for: py/build_whl.sh

@@ -5,6 +5,7 @@
 cd /workspace/Torch-TensorRT/py
 
 export CXX=g++
+export CUDA_HOME=/usr/local/cuda-11.3
 
 build_py36() {
     /opt/python/cp36-cp36m/bin/python -m pip install -r requirements.txt
@@ -33,4 +34,4 @@ build_py39() {
 build_py36
 build_py37
 build_py38
-build_py39
+build_py39