在開始學習深度學習之前,首先必須先把環境設定好,而Pytorch作為深度學習密不可分的項目,因此今天我們要來說說這個東西要怎麼安裝,在安裝之前,要先對自己的電腦有初步的認識,確定自己顯卡的配置才能正確安裝適合自己的版本,而這個環節如果出了問題,就會影響到後面的實作,接著讓我們開始吧!
Cmd
中輸入指令nvidia-smi
可查看目前的顯卡驅動版本,如果版本低於396.26請至官網更新驅動。Windows
系統,在這邊我選擇使用pip
套件進行安裝,語言選擇Python
,請確認裝置中是否有Nvidia的顯示卡,才可選擇CUDA版本
,無顯卡則選擇使用CPU
,使用CUDA可使訓練效率較快。
3. 接著在cmd
中輸入以下指令。
不過pytorch官方並沒有提出使用虛擬環境安裝的方式(也有可能是筆者沒有找到),因此在這邊我們依舊使用pip來安裝pytorch。
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
接著輸入y
安裝套件。
出現done
代表安裝完成
可使用簡單的程式測試一下pytorch的功能,在cmd
中先輸入python
後,接著輸入下列程式。
import torch
x = torch.rand(5, 3)
print(x)
輸出會類似於下圖
也可使用下列指令確定GPU驅動與CUDA是否啟用成功。
import torch
torch.cuda.is_available()
出現True就代表啟用成功了,那今天就到這結束,明天見。