PyTorch的简介和安装

PyTorch的简介和安装

1.1 PyTorch简介

PyTorch是由Facebook人工智能研究小组开发的一种基于Lua编写的Torch库的Python实现的深度学习库。

1.2 PyTorch的安装

• 进入PyTorch官网：https://pytorch.org/，选择对应版本进行安装

• 检验是否安装成功

  进入所在的虚拟环境，紧接着输入python，在输入下面的代码。

  import torch
  
  torch.cuda.is_available()
  False
  

  这条命令意思是检验是否可以调用cuda，如果我们安装的是CPU版本的话会返回False，能够调用GPU的会返回True。一般这个命令不报错的话就证明安装成功。

conda 虚拟环境

查看现存虚拟环境

查看已经安装好的虚拟环境

conda env list  

创建虚拟环境

在深度学习和机器学习中，我们经常会创建不同版本的虚拟环境来满足我们的一些需求。下面我们介绍创建虚拟环境的命令。

conda create -n env_name python==version

安装包

conda install package_name

卸载包

conda remove package_name

显示所有安装的包

conda list

删除