【Linux】Ubuntu20.04版本配置pytorch环境2023.09.05【教程】
在Ubuntu 20.04上配置PyTorch环境，你可以使用Anaconda进行快速安装。以下是步骤和示例代码：
- 如果尚未安装Anaconda，请从其官网下载并安装：https://www.anaconda.com/products/distribution
- 打开终端。
- 创建一个新的conda环境并激活它：
conda create -n pytorch_env python=3.8
conda activate pytorch_env- 使用官方PyTorch安装脚本安装PyTorch：
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch注意：上述命令安装的是与CUDA 11.3兼容的PyTorch版本。如果你的系统没有安装CUDA 11.3，你可以选择不安装cudatoolkit，或者安装CPU版本的PyTorch：
conda install pytorch torchvision torchaudio cpuonly -c pytorch- 验证PyTorch安装是否成功：
import torch
print(torch.__version__)如果你看到了版本号输出，说明PyTorch已成功安装。
评论已关闭