贴几个链接:
ngc pytorch容器版本对应关系
CUDA Toolkit版本及可用PyTorch对应关系
NVIDIA英伟达GPU显卡算力表
总结
安装torch时会安装对应的cuda版本,目前来看cuda10对应的是7.5的算力,适用于GPU20系列卡及以下;cuda11对应8.5的算法,适用于GPU30系列卡及以上(A100、A40等)
贴几个链接:
ngc pytorch容器版本对应关系
CUDA Toolkit版本及可用PyTorch对应关系
NVIDIA英伟达GPU显卡算力表
总结
安装torch时会安装对应的cuda版本,目前来看cuda10对应的是7.5的算力,适用于GPU20系列卡及以下;cuda11对应8.5的算法,适用于GPU30系列卡及以上(A100、A40等)