pyTorch环境搭建
Anaconda下载及安装
1.下载
- 官网下载(速度可能较慢)
- 清华源镜像下载(推荐)
2.安装
- 傻瓜式安装,一直next即可安装完成
- 安装完成后在电脑开始界面搜索Anaconda Prompt,打开后如下图,前面出现base,代表Anaconda安装成功!
Conda的解释
不同项目可能会用到pytorch
的不同版本,Anaconda
集成的Conda
包解决了这个问题,它可以创建两个"屋子",相互隔离,你要用哪个版本就进哪个"屋子"。(base代表初始的基础"屋子")
PyTorch的环境配置
- 创建名为
pytorch
的虚拟环境(“屋子”)
注意:电脑里要提前配有
python
环境(这里是python3.6,读者可根据自己的环境版本自行修改)
conda create -n pytorch python=3.6
- 环境创建完毕
- 启动
pytorch
环境:
conda activate pytorch
- 关闭当前环境:
conda deactivate
安装Pytorch
- 登录官网后,按如下配置:
- 复制命令到anaconda prompt终端,运行命令。
- 检验是否安装成功(未报错即安装成功)
- 查看
pytorch
是否能调用显卡(显卡必须支持cuda
),正常是返回ture,这里是因为是AMD显卡,天然不支持cuda
。
写在后面
显卡对于深度学习并不是很重要,没有显卡没有太大影响,显卡的作用是在模型训练时,让模型训练的更快,没有显卡会慢一点点而已。