PyTorch安装简介

一、Pip和Conda

Pip

pip pip是属于python的包管理工具。

Conda

Conda官网 conda是一个语言无关的开源包和环境管理工具。

一个区别

pip和conda的一个区别是:conda可以管理环境,而pip不可以;就python环境来说,conda通过命令就可以方便创建、切换和删除python环境,方便用户使用。若没有conda管理,用户需要下载不同版本的python到电脑上,切换环境不方便。

配合使用

pip和conda都可以安装包,两者配合使用较佳。

二、 Anaconda简介

Anaconda是一个集成平台,它集成了conda环境,自然也具备conda的功能。

Anaconda安装

Anaconda官网下载

常用操作

在开始菜单打开Anaconda Prompt
创建新的python环境

//conda create -n env_name python=version
eg. conda create -n pytorch python=3.8

查看环境列表,可以查看到base环境信息和刚才创建的环境名称(pytorch)及地址(Anaconda安装目录\envs\pytorch)

conda env list

激活环境

//conda activate 环境名
eg. conda activate pytorch

查看已安装包的列表

conda list

在pytorch环境下安装jupyter notebook

conda install nb_conda

安装成功后,在命令行输入 jupyter notebook打开notebook。
(我打开时报错500,命令行提示错误ImportError: cannot import name ‘contextfilter’ from ‘jinja2’,安装jinja2解决问题,pip install jinja2==3.0.3
报错处理参考)
退出当前环境

conda deactivate

三、 PyTorch安装

PyTorch官网

首先查看自己是否有CUDA:
无CUDA:Compute Platform选择cpu就可以了。

有CUDA:查看CUDA版本,根据显卡版本安装Pytorch。

命令行输入:nvidia-smi查看GPU状态,可以看到CUDA版本(我的10.1)

conda安装Python包 conda安装ipython_conda安装Python包

在开始菜单打开Anaconda Prompt,选择环境激活,开始安装:

可以选择Conda或Pip方式安装

conda安装Python包 conda安装ipython_pytorch_02


若官网首页没有你想要的版本,可以选择历史版本。

我的CUDA版本是10.1,所以要去找历史版本。

PyTorch历史版本下载 使用conda安装

conda安装Python包 conda安装ipython_pytorch_03

使用pip安装

conda安装Python包 conda安装ipython_conda安装Python包_04


若安装较慢,可以配置镜像源。

检测PyTorch是否安装成功?

// python
import torch
torch.cuda.is_available() //True 安装成功

torch.cuda.is_available()返回False

我最开始使用Conda方式安装的,torch.cuda.is_available()返回False,改用Pip安装就返回True了。

四、参考

conda 和 pip 的具体区别Anaconda介绍、安装及使用教程