Ubuntu下PyTorch能做深度學習嗎

在Ubuntu系統下，PyTorch可以用于深度學習。以下是詳細的安裝步驟和驗證方法：

安裝步驟

使用pip安裝PyTorch

1. 更新系統包：

sudo apt update
sudo apt upgrade

2. 安裝Python和pip（如果尚未安裝）：

sudo apt install python3 python3-pip

3. 創建虛擬環境（可選）：

python3 -m venv pytorch_env
source pytorch_env/bin/activate

4. 安裝PyTorch：

• CPU版本：

pip3 install torch torchvision torchaudio

• CUDA 11.3版本（如果你有NVIDIA GPU并且已經安裝了CUDA 11.3）：

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

• CUDA 11.7版本：

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

使用conda安裝PyTorch

1. 安裝Anaconda或Miniconda（如果尚未安裝）：

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh

2. 創建并激活conda環境（可選）：

conda create -n pytorch_env python=3.8
conda activate pytorch_env

3. 安裝PyTorch：

• CPU版本：

conda install pytorch torchvision torchaudio cpuonly -c pytorch

• CUDA 11.3版本：

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

• CUDA 11.7版本：

conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch

驗證安裝

安裝完成后，可以通過以下命令驗證PyTorch是否安裝成功：

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 如果安裝了CUDA版本，應該返回True

通過以上步驟，你可以在Ubuntu系統上成功安裝PyTorch，并使用它進行深度學習任務。如果在安裝過程中遇到問題，可以參考PyTorch官方文檔或相關社區論壇尋求幫助。