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大洋,
下載完畢,大致是這個樣子
下面 開始安裝