主要原因應該是硬件能夠支持的算力比較高,能達到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刪除。