要檢查在 Linux 上是否已成功安裝 PyTorch,可以按照以下步驟進行:
打開終端:首先,打開你的 Linux 終端。
啟動 Python 或 Python3 解釋器:
python
或 python3
并按回車鍵,進入 Python 交互環境。具體使用哪個命令取決于你的系統配置。導入 PyTorch:
import torch
檢查 PyTorch 版本:
print(torch.__version__)
驗證 GPU 支持(可選):
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)
cuda
,則表示 PyTorch 可以使用 GPU。你還可以進一步檢查可用的 GPU 數量:print(torch.cuda.device_count())
測試簡單的張量操作:
x = torch.rand(2, 3)
y = torch.rand(2, 3)
z = x + y
print(z)
如果在以上步驟中沒有出現錯誤消息,并且版本號和設備信息顯示正確,那么 PyTorch 應該已經在你的 Linux 系統上成功安裝。如果在導入或運行過程中遇到錯誤,請檢查你的安裝步驟是否正確,或者查看 PyTorch 官方網站獲取更多支持和安裝指南。