主要原因应该是硬件能够支持的算力比较高,能达到8.6,但是cuda11.0支持不了这么高的算力,通过下述脚本,设置环境变量,降低算力要求,即可:
export TORCH_CUDA_ARCH_LIST="7.5"
主要原因应该是硬件能够支持的算力比较高,能达到8.6,但是cuda11.0支持不了这么高的算力,通过下述脚本,设置环境变量,降低算力要求,即可:
export TORCH_CUDA_ARCH_LIST="7.5"
本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。