PyTorch在Ubuntu上的使用教程有哪些

在Ubuntu上安装和使用PyTorch的教程有很多，以下是一些常用的教程和资源：

使用pip安装PyTorch

1. 更新系统软件包：

sudo apt update

2. 安装Python3-venv：

sudo apt install python3-venv

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. 验证安装：

import torch
print(torch.__version__)

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

pip3 install torch torchvision torchaudio

使用Anaconda安装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 create -n pytorch python3.8
conda activate pytorch

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

conda install pytorch torchvision torchaudio cpuonly -c pytorch

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

conda install pytorch torchvision torchaudio pytorch-cuda11.7 -c pytorch -c nvidia

6. 验证安装：

import torch
print(torch.__version__)

使用CUDA和cuDNN安装PyTorch

1. 安装NVIDIA显卡驱动：

sudo apt install nvidia-cuda-toolkit nvidia-driver-510

2. 安装CUDA：

wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run
sudo sh cuda_11.1.1_455.32.00_linux.run

3. 安装cuDNN：

sudo cp cuda/include/cudnn.h /usr/local/cuda-11.1/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.1/lib64
sudo chmod a+r /usr/local/cuda-11.1/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.1/lib64/libcudnn*

4. 安装PyTorch（支持GPU）：

pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/torch_stable.html

希望这些教程能帮助你成功在Ubuntu上安装和使用PyTorch！