如何用conda安装支持CUDA11.6的GPU版PyTorch 1.11？

想要用conda安装支持特定CUDA版本（比如11.6）的GPU版PyTorch 1.11。已知安装最新版本的命令如下：

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

而旧版本无GPU支持的PyTorch安装方法为：

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 -c pytorch

请问以下命令能否通过conda安装带GPU支持的PyTorch 1.11？

conda install pytorch==1.11.0 torchvision==0.12.0 pytorch-cuda=11.6 -c pytorch -c nvidia

不行，这个命令无法成功安装支持CUDA 11.6的GPU版PyTorch 1.11.0，核心原因有两点：

• pytorch-cuda包的版本适配限制：pytorch-cuda是PyTorch 1.12及以后版本才推出的conda包，用来简化CUDA依赖的管理，PyTorch 1.11.0的conda仓库没有和这个包做关联配置，没法通过它给PyTorch 1.11.0绑定CUDA环境。
• PyTorch 1.11.0本身不支持CUDA 11.6：PyTorch 1.11.0发布时，官方预编译的GPU包最高只支持CUDA 11.3，根本没有适配CUDA 11.6的版本。

如果要安装GPU版PyTorch 1.11.0，正确的做法是用cudatoolkit指定官方支持的CUDA版本（可选10.2或11.3），命令如下：

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

内容的提问来源于stack exchange，提问作者Meysam Sadeghi