如何檢查Linux PyTorch安裝是否成功

要確認(rèn) linux 系統(tǒng)中 pytorch 是否安裝成功，可以參考以下操作流程：

1. 打開終端： 啟動你的 Linux 終端程序。

2. 進(jìn)入 python 環(huán)境： 根據(jù)系統(tǒng)設(shè)置，輸入 python 或 python3 命令來啟動 Python 交互界面。

    python

   或者使用

    python3

3. 嘗試導(dǎo)入 PyTorch 模塊： 在 Python 中試著加載 torch 和 torchvision 模塊，以驗證其可用性。

    import torch  import torchvision

4. 查看當(dāng)前安裝的 PyTorch 版本： 輸出 PyTorch 的版本號，用于確認(rèn)安裝狀態(tài)。

    print(torch.__version__)

5. 檢測 GPU 驅(qū)動支持情況（如適用）： 如果你擁有 NVIDIA 顯卡并已配置 CUDA 環(huán)境，可以通過下面的代碼判斷 PyTorch 能否識別到顯卡設(shè)備。

    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")  print(device)

   若輸出為 cuda，則表示 PyTorch 成功識別到了 GPU。

6. 執(zhí)行基礎(chǔ)張量運(yùn)算測試： 最后，運(yùn)行一個簡單的張量創(chuàng)建和打印操作，進(jìn)一步驗證功能是否正常。

    x = torch.rand(2, 3)  print(x)

   正常情況下會顯示一個 2 行 3 列的隨機(jī)數(shù)值矩陣。

只要在完成以上所有步驟時未出現(xiàn)異常，并且能夠正確獲取版本與設(shè)備信息，就說明 PyTorch 已經(jīng)順利安裝并可在你的 Linux 系統(tǒng)上使用。