通过Anaconda安装PyTorch时适配的CUDA版本如何选择？

你给出的conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge命令完全可以正常使用，不会出现适配问题，原因如下：

• PyTorch官方预编译的CUDA 11.1版本安装包，默认兼容所有计算能力≥3.5的NVIDIA显卡，你的GTX 1660 Ti计算能力为7.5，远高于最低适配要求，属于完全兼容的硬件范围。
• CUDA版本本身是向下兼容低计算能力显卡的，你看到的8.7是目前NVIDIA显卡的最高计算能力规格，不代表CUDA 11.1只能适配最高计算能力的显卡，两者的参数差不会带来适配问题。
• 不需要额外单独安装官方的CUDA Toolkit本体，conda安装过程中自带的cudatoolkit已经包含了PyTorch运行需要的全部依赖，你只需要保证本地NVIDIA显卡驱动版本≥456.38（CUDA 11.1要求的最低驱动版本）即可正常运行。

安装完成后打开Python交互环境，运行以下代码即可验证GPU是否正常调用：

import torch
print(torch.cuda.is_available())

输出为True就说明适配完全正常。

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