QDQBert example update (#16395)

* update Dockerfile and utils_qa

* Update README.md

examples/research_projects/quantization-qdqbert/Dockerfile

@@ -12,7 +12,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-FROM nvcr.io/nvidia/pytorch:21.07-py3
+FROM nvcr.io/nvidia/pytorch:22.02-py3
 LABEL maintainer="Hugging Face"
 LABEL repository="transformers"
 
@@ -20,11 +20,7 @@ RUN apt-get update
 RUN apt-get install sudo
 
 RUN python3 -m pip install --no-cache-dir --upgrade pip
-RUN python3 -m pip install --no-cache-dir --ignore-installed ruamel.yaml \
-    mkl \
-    absl-py \
-    yamlpy \
-    tensorboardX
+RUN python3 -m pip install --no-cache-dir --ignore-installed pycuda
 RUN python3 -m pip install --no-cache-dir \
     pytorch-quantization --extra-index-url https://pypi.ngc.nvidia.com
 
@@ -34,4 +30,4 @@ RUN cd transformers/ && \
     python3 -m pip install --no-cache-dir .
 
 RUN python3 -m pip install --no-cache-dir datasets \
-    accelerate
+    accelerate