基于python的PyTorch的学习(第一期)

168 阅读1分钟

pyTorch环境搭建

Anaconda下载及安装

1.下载

捕获.PNG

捕获.PNG

2.安装

  • 傻瓜式安装,一直next即可安装完成

image.png

  • 安装完成后在电脑开始界面搜索Anaconda Prompt,打开后如下图,前面出现base,代表Anaconda安装成功!

捕获.PNG

Conda的解释

不同项目可能会用到pytorch的不同版本,Anaconda集成的Conda包解决了这个问题,它可以创建两个"屋子",相互隔离,你要用哪个版本就进哪个"屋子"。(base代表初始的基础"屋子")

PyTorch的环境配置

  1. 创建名为pytorch的虚拟环境("屋子")

注意:电脑里要提前配有python环境(这里是python3.6,读者可根据自己的环境版本自行修改)

conda create -n pytorch python=3.6

捕获.PNG

  1. 环境创建完毕

捕获.PNG

  • 启动pytorch环境:
conda activate pytorch
  • 关闭当前环境:
conda deactivate

安装Pytorch

  1. 官网网址

image.png

  1. 登录官网后,按如下配置:

捕获.PNG

  1. 复制命令到anaconda prompt终端,运行命令。

捕获.PNG

  1. 检验是否安装成功(未报错即安装成功)

捕获.PNG

  1. 查看pytorch是否能调用显卡(显卡必须支持cuda),正常是返回ture,这里是因为是AMD显卡,天然不支持cuda

image.png

写在后面

显卡对于深度学习并不是很重要,没有显卡没有太大影响,显卡的作用是在模型训练时,让模型训练的更快,没有显卡会慢一点点而已。