深入浅出Pytorch系列(1):安装
为了更好的学习深度学习,动手搭建模型,Pytorch是不可不学的工具!
Pytorch的安装
从Anaconda入手进行安装Pytorch较为简单!具体步骤如下:
Step 1:登陆Anaconda | Individual Edition,选择相应系统配置进行下载;
Step 2:在开始页找到Anaconda Prompt,一般在Anaconda3的文件夹下;
Step 3:创建虚拟环境
在Anaconda Prompt下输入命令:
conda env list‘’‘’
查看虚拟环境;
输入命令:
conda create -n name python==3.8
Step 4:登录官网Pytorch官网
Step 5:点击Install按钮,进入相关界面
Step 6:结合自己情况选择命令并复制下来,然后使用conda下载
Step 6:检验是否安装成功
进入所在的虚拟环境,输入
import torch
torch.cuda.is_available()
这条命令意思是检验是否可以调用cuda,如果安装的是CPU的小伙伴们会返回False,GPU的小伙伴会返回true。一般这个命令不报错的话就证明安装成功。
注:本人安装在WINDOWS系统,且过程中未涉及‘换源’。