python包:torch

torch  這個包非常大,安裝也非常麻煩。

基本可以上,他是深度學習裏面重要的一個包,

要安裝torch   首先要確定自己的操作系統, 自己的python 版本, 自己的版本,  是要安裝CPU版本的 torch, 還是GPU版本的torch 

 

***

我這邊是WIN10專業版。Python 3.8.10 。

我自己有泰坦RTX的顯卡,所以肯定要裝GPU版本的 ,不然浪費硬件資源,

通過指令:cuda --version

我發現我的cuda 版本是:12.2

 

下面就可以找安裝包了(就是那個whl 文件)

 

https://download.pytorch.org/whl/torch/

當下時間(20240424 1136)找不到cuda12.2  ,最新的cuda121 ,所以先用它, 大概率沒問題

 

 

開源的東西,版本太多了,一個torch  就有1833個版本 ,幹! 

這個torch文件 有2G大小 ,在國內的網絡環境下, 不實用鏡像資源 直接pip 安裝 大概率完蛋

所以我這邊打算下載whl 文件, 然後本地安裝, 即便是如此,因爲大量依賴項的原因,估計還是會下載 東西

 

這個安裝太複雜了怪不得網上  遠程裝 AI環境,要收費100大洋,

 

 

下載完畢,大致是這個樣子

 

下面 開始安裝

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章