Pytorch安装步骤详解
conda环境
Anaconda下载地址:https://www.anaconda.com/download/
本文操作系统默认的是windows,所以后续的操作都是针对windows系统上进行的,如果电脑上没有安装Anaconda并且对Anaconda不熟悉的读者,请参考这篇文章:Anaconda介绍、安装及使用教程
下面我们先来验证一下自己电脑有没有conda环境吧,其实操作很简单,使用快捷键win+R输入cmd启动命令行:
在命令行界面输入conda并回车出现以下内容就表示电脑上已经存在conda环境了:
如果出现这个界面,那么恭喜你可以在你自己的电脑上开始安装pytorch了
安装Pytorch
在实际生产开发中,我们往往会为每个项目单独配一个环境,这个时候原生的python环境已经不能满足我们的要求了,就需要使用conda环境为项目创建单独的环境,而且conda还会自动为你安装pytorch需要的cuda和cudnn,为后续自己安装省去了很多的麻烦。
首先我们打开pytorch官网:https://pytorch.org/
进入官网后在首页下方找到以下界面并复制红框中的命令:
如果你的电脑上没有gpu,那么就选择cpu的命令进行安装即可。python的版本没有限制,用conda安装的时候python3.9可能需要一些额外的配置,按照命令提示安装即可。我们这里选择用pip进行安装。
在命令行创建conda环境
1.打开命令行,输入以下命令创建:
conda create -n pytorchn python=3.6
这里的pytorchn是你创建环境的名称,python=3.6是你创建的python版本。
输入上述命令,在出现提示的时候输入y安装即可,出现以下提示表示安装成功:
2.创建好环境后,我们需要输入conda activate pytorchn来进入你刚刚创建的环境中:
在创建好的环境中把刚刚在官网复制的命令输入命令行安装即可:
因为文件较大,耐心等待安装即可。
3.若安装成功后,在pycharm中把环境切换成你刚刚创建的conda环境中并在文件中输入
import torch
红框中的代码不报红则表示安装成功了
还可以输入以下代码查看torch的版本:
In[2]:import torch
In[3]:torch.__version__
Out[3]: '1.7.0'
至此,pytorch已在你的电脑上安装成功了!