Ubuntu下PyTorch能用于深度学习吗

是的，PyTorch可以在Ubuntu下用于深度学习。以下是在Ubuntu上安装PyTorch的步骤：

使用pip安装PyTorch

1. 更新系统软件包：

sudo apt update

2. 安装Python3-venv：

sudo apt install python3-venv -y

3. 创建并激活虚拟环境：

mkdir pytorch_env
cd pytorch_env
python3 -m venv pytorch_env
source pytorch_env/bin/activate

4. 安装PyTorch（仅支持CPU）：

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

5. 验证安装：

python -c "import torch; print(torch.__version__)"

使用conda安装PyTorch

1. 更新系统软件包：

sudo apt update

2. 安装Anaconda：

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

3. 创建并激活新的conda环境：

conda create -n myenv
conda activate myenv

4. 安装PyTorch（支持GPU，需要先安装CUDA）：

conda install pytorch torchvision torchaudio cudatoolkit10.2 -c pytorch

5. 验证安装：

python -c "import torch; print(torch.__version__)"
python -c "import torch; print(torch.cuda.is_available())"

通过以上步骤，您可以在Ubuntu系统上成功安装PyTorch，并开始进行深度学习项目。