Pytorch 环境搭建

102 阅读1分钟

PyTorch的安装十分简单,根据PyTorch官网,对系统选择和安装方式等灵活选择即可。 这里以anaconda为例,简单的说一下步骤和要点。 国内安装anaconda建议使用清华镜像。

安装Pytorch

  • 进入官网,选择安装类型

image.png

  • 复制命令到命令行执行
conda install pytorch torchvision torchaudio cpuonly -c pytorch

执行结果如下:

image.png

测试

image.png