一、pytorch安装配置
1、pytorch 环境配置
首先安装Anaconda,包含了大量的工具包。不必下载最新版本,防止出现问题。
https://repo.anaconda.com/ 安装3.6版本。记住安装路径,跳过安装VScode.打开anaconda promote.出现base即可。
2、显卡准备(训练加速作用)
pytorch 和tensflow主要支持NVIDIA,显卡配置涉及驱动+CUDA工具包(跟随pytorch一键安装)
任务管理器->性能->GPU是否正常显示信号。
3、管理环境(有序)
conda create -n pytorch python=3.6 安装 一直输入y 。用conda activate pytorch激活环境
用pip list 显示列表,无torch则下载
4、安装pytorch
选择稳定版本 1.3 windows下选择conda linux下选择pip安装包
输入命令,pip list看包列表有无。或(慢)复制安装包放到anaconda目录的pkgs,再安装。
5、检验pytorch是否安装成功安装
torch是否能用GPU命令:先进入pytorch,然后import torch 没有出现问题。torch安装成功
然后输入torch.cuda.is_available()回复True则能用本机GPU