本地安裝Cuda版本的Torch
之前安裝了Torch,但是是CPU版本的,對於模型訓練而言cpu版本的Torch實在是太慢了,所以開始安裝Cuda(GPU)版本的torch。
推薦下載站:上海交通大學鏡像
為什么要本地安裝Torch
由於現在國內直接安裝Torch還是太慢了,主要是國內直連國外的服務器可能還是比較慢的。而且,如果實在服務器上安裝Torch,聯網實在是一件麻煩的事情。所以本地安裝其實更加高效方便,不過在剛開始的時候可能更加麻煩一點。
Torch的必備要求
- Nvidia版本支持Cuda的顯卡:可以在系統配置里面直接看到,這里不贅述了。
- 安裝
Python
或者C++
:這里我選擇的是Python,畢竟現在做深度學習還是Python居多,其中很多的運算密集型操作也可通過調用Cpp的包完成。 - 安裝和顯卡
Cuda
版本對應的英偉達Cuda驅動:可以前往英偉達官網下載驅動。 - 下載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!