Pytorch超简单安装教程

安装Pytorch

1.1 安装Annaconda

安装Pytorch首先需要安装Annaconda,按照教程,安装了Annaconda5.2.0的版本。 记住安装路径即可,其余均选默认。

1.2 安装Pytorch

第一步,打开Anaconda Prompt 。然后输入

conda create -n pytorch python=3.6

第二步,选择y,将安装包安装完成。 第三步,输入activate pytorch,自动进入pytorch环境。 第四步,输入pip list,列出列表,查看是否有pytorch,不安装的话肯定没有。。 第五步,打开,点击Get Started。 有GPU的话,按照显卡选择CUDA版本。我自己选择10.2的。 然后再prompt里输入

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

运行完后输入y即可自动安装。 如果网速差的话可以先下载,再安装。见 第六步,验证一下是否安装成功。

1.3 安装Pycharm

Pycharm是目前比较流行的一款Python编辑器。 安装路径如下: 下载社区版即可,安装路径建议选C盘以外的位置。 注意,在安装选项里只选择.py即可。 1.打开pycharm,选择新建项目。 interpreter选择python.exe,选择create. 2. 打开python Console. 3. 输入import torch,如果没报错,说明导入pytorch成功。

1.4 Jupyter

Jupyter是安装Annaconda自带的编译器,是网页版的。 选择在pytorch环境安装jupyter。

  1. 打开prompt,安装jupyter安装包。
conda activate pytorch
conda install nb_conda

2.点击y. 3.安装完毕后输入Jupyter notebook,即可进入jupyter内。 4.验证。

经验分享 程序员 微信小程序 职场和发展