主要原因应该是硬件能够支持的算力比较高，能达到8.6，但是cuda11.0支持不了这么高的算力，通过下述脚本，设置环境变量，降低算力要求，即可：
export TORCH_CUDA_ARCH_LIST="7.5"
相关文章：
相似解决方案
热门标签
主要原因应该是硬件能够支持的算力比较高，能达到8.6，但是cuda11.0支持不了这么高的算力，通过下述脚本，设置环境变量，降低算力要求，即可：
export TORCH_CUDA_ARCH_LIST="7.5"
相关文章：