摘要: 主要原因应该是硬件能够支持的算力比较高,能达到8.6,但是cuda11.0支持不了这么高的算力,通过下述脚本,设置环境变量,降低算力要求,即可:export TORCH_CUDA_ARCH_LIST="7.5" 阅读全文
posted @ 2021-02-08 23:25 Wind·Chaser 阅读(2305) 评论(0) 推荐(0) 编辑