pip 安装pytorch
安装pytorch方式有两种:用anaconda安装或者pip方式,此笔记是pip安装方式。
根据pytorch官网我们需要做的准备工作
下载python和搞清楚cuda
1.下载python,安装的时候要记得勾选添加到环境,如果没有勾选需要自己添加环境。安装完python后,pip也安装完了。(可以根据以下步骤查看pip是否安装成功:win+R——输入cmd——输入pip --version——如有版本号就说明安装成功)
2.cuda版本的选择要根据自己电脑配置进行选择,以我自己台式电脑为例。
1.如果有显卡就说明可以安装cuda,查看自己是否有显卡步骤:控制面板——NVIDIA控制面板(如果找不到就点右边查看方式选择小图标)——系统信息(在面板左下方)——组件(我的是cuda10.1)。