Pytorch GPU版本的安裝

先查看自己電腦英偉達CUDA支持的版本信息

• 點幫助——系統信息：
  
• 或者cmd查一下
  

下載相關CUDA驅動

• CUDA下載：https://developer.nvidia.com/cuda-toolkit-archive

• CUDNN下載：https://developer.nvidia.com/rdp/cudnn-download

• 登錄后下載
  
  
  

• 以管理員身份運行安裝
  
  
  
  

• 如果當前版本比新版本小，那就打勾繼續下一步，否則把勾去掉：
  
  
  繼續安裝
  
  

• 測試一下安裝成功沒
  
  有這個版本就成功了

cudnn

• 把另一個壓縮包解壓，點進去重命名
  
• 然后復制到剛剛安裝的這個v10.2里面
  

• 配置環境變量
  加這兩個就行了
  

pytorch GPU版本的安裝

• pytorch下載：pytorch.org
  

• 復制命令執行即可，要是速度慢或者中間報錯，就加個鏡像源給他
  
  

• 安裝后測試一下
  

• 說是少這個，那就裝一下，一般都沒有提示這個的，應該是我重裝系統后這個東西沒有了
  

• 重啟
  
  如果輸出True，那就是安裝成功且可以使用gpu加速了

• 如果是用conda安裝，加速可以參考下面鏈接
  - https://zhuanlan.zhihu.com/p/144575524
  - https://blog.csdn.net/qq_41936559/article/details/102699082

• 結果
  
  如果輸出True，那就是安裝成功且可以使用gpu加速了
  
  

=完美=