Introduced --torch-with-cuda11 option for 3rd party sanity tests

[ljaljushkin]

Changes

Introduced --torch-with-cuda11 option for 3rd party sanity tests to install torch compatible with GTX3090.

Reason for changes

There's issue with torch installation inside a 3rd party test.

The 3rd party sanity test creates a virtual python environment by itself. Particularly, it installs default pytorch 1.9.1 with CUDA 10 support.

BKC_TORCH_VERSION=1.9.1
pip_runner.run_pip("install torch=={}".format(BKC_TORCH_VERSION))

The target hardware might have RTX-3090 cards, which requires torch with CUDA 11 support. It can be installed as follows:

pip install torch==1.9.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html 

Currently, the corresponding CI job has RTX3090 card now. As results, test fails with the following error:

NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation

The test is not hardware-agnostic. It doesn't know how to install pytorch properly. That's why some option is required.

Related tickets

89154

Tests

• [ ] need to make sure that 3rd sanity test passes with the option
• [ ] modify scripts in Jenkins to pass the option