pytorch——gpu版本安裝

1、檢查顯卡驅(qū)動（控制面板-小圖標-英偉達，幫助-信息），安裝和顯卡對應的cuda版本，添加環(huán)境變量

用cudann里面的文件替換，參考地址：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

（可以在沒有conda環(huán)境下直接使用）

建新的anaconda環(huán)境

（檢查顯卡驅(qū)動是否滿足cuda版本，torch版本 和 cuda版本需要對應）

（可以在清華源下載離線安裝包，https://mirrors.tuna.tsinghua.edu.cn/help/pypi/）

分步安裝，注意區(qū)分linux 和 win版本

【1】https://www.cnblogs.com/lyl0618/p/13323010.html? ??torch& torchvision對應版本

最后編輯于 ：
?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成，瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點，簡書系信息發(fā)布平臺，僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容