Torch的離線安裝


本地安裝Cuda版本的Torch

之前安裝了Torch,但是是CPU版本的,對於模型訓練而言cpu版本的Torch實在是太慢了,所以開始安裝Cuda(GPU)版本的torch。

推薦下載站:上海交通大學鏡像

為什么要本地安裝Torch

由於現在國內直接安裝Torch還是太慢了,主要是國內直連國外的服務器可能還是比較慢的。而且,如果實在服務器上安裝Torch,聯網實在是一件麻煩的事情。所以本地安裝其實更加高效方便,不過在剛開始的時候可能更加麻煩一點。

Torch的必備要求

  1. Nvidia版本支持Cuda的顯卡:可以在系統配置里面直接看到,這里不贅述了。
  2. 安裝Python或者C++:這里我選擇的是Python,畢竟現在做深度學習還是Python居多,其中很多的運算密集型操作也可通過調用Cpp的包完成。
  3. 安裝和顯卡Cuda版本對應的英偉達Cuda驅動:可以前往英偉達官網下載驅動。
  4. 下載Torch的wheel文件,可以選擇上面提供的下載站下載,也可以自行尋找whl文件下載.

Caution: 下載時可以使用vpn,安裝時則不能使用vpn,否則可能出現Hostname報錯

graph TD; A[確認你的GPU支持cuda]-->B[查看自己的Cuda版本]; B-->C[根據自己的Cuda版本下載相關的英偉達官網cuda包]; C-->D[確認完成Python或者Cpp的下載]; D-->E[根據自己的Python版本和Cuda版本前往下載站下載torch.whl文件]; E-->F[pip或者conda直接安裝,此時不能連接vpn.同時需要確保torch.whl的名稱符合標准命名方式];

Success

Ooops, it works!


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM