CUDA 11.6对应的PyTorch版本是1.7.1。

是的，CUDA 11.6对应的PyTorch版本是1.7.1。你可以在PyTorch官方网站上找到与你的CUDA版本匹配的PyTorch版本。另外，也可以通过conda或pip安装命令安装特定版本的PyTorch。例如，使用conda安装PyTorch 1.7.1并使用CUDA 11.6，可以执行以下命令：

conda install pytorch=1.7.1 torchvision torchaudio cudatoolkit=11.6 -c pytorch

如果你使用pip安装PyTorch 1.7.1并使用CUDA 11.6，可以执行以下命令：

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/cu110/torch_stable.html

其中，cu110代表CUDA 11.0，你需要将其替换为cu116，以匹配你的CUDA版本。