PyTorch在Ubuntu上的安裝步驟

在Ubuntu上安裝PyTorch的步驟如下：

安裝前的準備工作

1. 確認顯卡：確認你的顯卡是否支持CUDA。如果是NVIDIA顯卡，請安裝相應的NVIDIA驅動。
2. 檢查CUDA版本：運行命令 nvidia-smi 來查看你的顯卡支持的CUDA版本。
3. Python環境：建議使用Python 3.7及以上版本，推薦使用Anaconda來管理環境。

使用Anaconda安裝PyTorch（推薦）

1. 創建虛擬環境：

conda create -n pytorch_env python=3.9

2. 激活虛擬環境：

conda activate pytorch_env

3. 從PyTorch官網獲取安裝命令：
   • 訪問 PyTorch官網
   • 根據你的實際情況選擇對應的配置（操作系統、包管理器、語言、CUDA等）。
4. 復制并粘貼安裝命令：

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

（請將 11.8 替換為你的CUDA版本）

使用pip安裝PyTorch

1. 通用版本（自動檢測CUDA）：

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

2. 指定版本安裝：

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

（請將 cu118 替換為你的CUDA版本）

驗證安裝

在Python中輸入以下代碼來驗證PyTorch是否成功安裝：

import torch
print(f"PyTorch版本: {torch.__version__}")
print(f"CUDA可用: {torch.cuda.is_available()}")
print(f"當前設備: {torch.device('cuda' if torch.cuda.is_available() else 'cpu')}")

注意事項

• 確保CUDA和cuDNN的版本與PyTorch的要求相匹配。
• 如果在安裝過程中遇到網絡問題，可以嘗試使用國內的鏡像源，如清華大學的鏡像源。

以上步驟應該可以幫助你在Ubuntu上成功安裝PyTorch。如果在安裝過程中遇到問題，可以參考PyTorch的官方文檔或相關社區論壇尋求幫助。