CUDA版本、显卡算力、Pytorch对应关系

背景

试用合作医院提供的堡垒机服务器，自带的NVIDIA-A6000显卡不支持结构像自动分割的pipeline镜像，运行时报错capability sm_86 is not compatible。发现是由于显卡的架构比较新，旧版pytorch库不支持。同时根据输出可以看到 The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70 sm_75当前pytorch只能支持上面几种（显卡算力）架构。

解决方案

把基础镜像直接更新到pytorch/pytorch:1.7.0-cuda11.0-cudnn8-runtime，解决。 NOTE: CUDA10.x最高支持算力7.x; CUDA11.0最高支持算力8.x